fix: fixed device issue
Browse files
app.py
CHANGED
@@ -72,11 +72,11 @@ def generate_audio(text, model_id, language):
|
|
72 |
if pipe_dict["language"] != language:
|
73 |
gr.Warning(f"Language has changed - loading new default model: {default_model_per_language[language]}")
|
74 |
pipe_dict["language"] = language
|
75 |
-
pipe_dict["original_pipe"] = pipeline("text-to-speech", model=default_model_per_language[language], device=
|
76 |
|
77 |
if pipe_dict["current_model"] != model_id:
|
78 |
gr.Warning("Model has changed - loading new model")
|
79 |
-
pipe_dict["pipe"] = pipeline("text-to-speech", model=model_id, device=
|
80 |
pipe_dict["current_model"] = model_id
|
81 |
|
82 |
num_speakers = pipe_dict["pipe"].model.config.num_speakers
|
|
|
72 |
if pipe_dict["language"] != language:
|
73 |
gr.Warning(f"Language has changed - loading new default model: {default_model_per_language[language]}")
|
74 |
pipe_dict["language"] = language
|
75 |
+
pipe_dict["original_pipe"] = pipeline("text-to-speech", model=default_model_per_language[language], device=device)
|
76 |
|
77 |
if pipe_dict["current_model"] != model_id:
|
78 |
gr.Warning("Model has changed - loading new model")
|
79 |
+
pipe_dict["pipe"] = pipeline("text-to-speech", model=model_id, device=device)
|
80 |
pipe_dict["current_model"] = model_id
|
81 |
|
82 |
num_speakers = pipe_dict["pipe"].model.config.num_speakers
|