fix: fixed device issue (#3)

- fix: fixed device issue (2fe6565b04bf41c0106d977714a658c39befdeaf)


Co-authored-by: Levi Wenger <[email protected]>

app.py

@@ -72,11 +72,11 @@ def generate_audio(text, model_id, language):
     if pipe_dict["language"] != language:
         gr.Warning(f"Language has changed - loading new default model: {default_model_per_language[language]}")
         pipe_dict["language"] = language
-        pipe_dict["original_pipe"] = pipeline("text-to-speech", model=default_model_per_language[language], device=0)
+        pipe_dict["original_pipe"] = pipeline("text-to-speech", model=default_model_per_language[language], device=device)
     
     if pipe_dict["current_model"] != model_id:
         gr.Warning("Model has changed - loading new model")
-        pipe_dict["pipe"] = pipeline("text-to-speech", model=model_id, device=0)
+        pipe_dict["pipe"] = pipeline("text-to-speech", model=model_id, device=device)
         pipe_dict["current_model"] = model_id
 
     num_speakers = pipe_dict["pipe"].model.config.num_speakers