Fabrice-TIERCELIN commited on
Commit
f87052e
·
verified ·
1 Parent(s): 0ba1c46

output_filename

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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="output.wav",
100
  ),
101
- "output.wav",
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="output.wav",
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: