output_filename
Browse files
app.py
CHANGED
@@ -75,7 +75,7 @@ def predict(prompt, language, gender, audio_file_pth, mic_file_path, use_mic):
|
|
75 |
language = "fr-fr"
|
76 |
if m.find("/fr/") != -1:
|
77 |
language = None
|
78 |
-
predict_on_gpu(prompt, speaker_wav, language)
|
79 |
except RuntimeError as e :
|
80 |
if "device-assert" in str(e):
|
81 |
# cannot do anything on cuda device side error, need to restart
|
@@ -96,19 +96,19 @@ def predict(prompt, language, gender, audio_file_pth, mic_file_path, use_mic):
|
|
96 |
|
97 |
return (
|
98 |
gr.make_waveform(
|
99 |
-
audio=
|
100 |
),
|
101 |
-
|
102 |
information,
|
103 |
)
|
104 |
|
105 |
@spaces.GPU(duration=60)
|
106 |
-
def predict_on_gpu(prompt, speaker_wav, language):
|
107 |
tts.tts_to_file(
|
108 |
-
text=prompt,
|
109 |
-
file_path=
|
110 |
-
speaker_wav=speaker_wav,
|
111 |
-
language=language
|
112 |
)
|
113 |
|
114 |
with gr.Blocks() as interface:
|
|
|
75 |
language = "fr-fr"
|
76 |
if m.find("/fr/") != -1:
|
77 |
language = None
|
78 |
+
predict_on_gpu(prompt, speaker_wav, language, output_filename)
|
79 |
except RuntimeError as e :
|
80 |
if "device-assert" in str(e):
|
81 |
# cannot do anything on cuda device side error, need to restart
|
|
|
96 |
|
97 |
return (
|
98 |
gr.make_waveform(
|
99 |
+
audio = output_filename,
|
100 |
),
|
101 |
+
output_filename,
|
102 |
information,
|
103 |
)
|
104 |
|
105 |
@spaces.GPU(duration=60)
|
106 |
+
def predict_on_gpu(prompt, speaker_wav, language, output_filename):
|
107 |
tts.tts_to_file(
|
108 |
+
text = prompt,
|
109 |
+
file_path = output_filename,
|
110 |
+
speaker_wav = speaker_wav,
|
111 |
+
language = language
|
112 |
)
|
113 |
|
114 |
with gr.Blocks() as interface:
|