huanne1002 commited on
Commit
ef1abc0
verified
1 Parent(s): 7a24b6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -180,9 +180,9 @@ def predict(
180
  torchaudio.save(buffer, torch.tensor(out["wav"]).unsqueeze(0), 24000, format='wav')
181
 
182
  buffer.seek(0)
183
- wav_base64 = base64.b64encode(buffer.read()).decode('utf-8')
184
- print(wav_base64)
185
- wav_base64
186
 
187
  except RuntimeError as e:
188
  if "device-side assert" in str(e):
@@ -248,7 +248,7 @@ def predict(
248
  "Something unexpected happened please retry again."
249
  )
250
  return (None, metrics_text)
251
- return ("output.wav", metrics_text)
252
 
253
 
254
  with gr.Blocks(analytics_enabled=False) as demo:
@@ -317,6 +317,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
317
  with gr.Column():
318
  audio_gr = gr.Audio(label="Synthesised Audio", autoplay=True)
319
  out_text_gr = gr.Text(label="Metrics")
 
320
 
321
  tts_button.click(
322
  predict,
@@ -326,7 +327,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
326
  ref_gr,
327
  normalize_text,
328
  ],
329
- outputs=[audio_gr, out_text_gr],
330
  api_name="predict",
331
  )
332
 
 
180
  torchaudio.save(buffer, torch.tensor(out["wav"]).unsqueeze(0), 24000, format='wav')
181
 
182
  buffer.seek(0)
183
+ base_gr = base64.b64encode(buffer.read()).decode('utf-8')
184
+ print(base_gr)
185
+ base_gr
186
 
187
  except RuntimeError as e:
188
  if "device-side assert" in str(e):
 
248
  "Something unexpected happened please retry again."
249
  )
250
  return (None, metrics_text)
251
+ return ("output.wav", metrics_text, base_gr)
252
 
253
 
254
  with gr.Blocks(analytics_enabled=False) as demo:
 
317
  with gr.Column():
318
  audio_gr = gr.Audio(label="Synthesised Audio", autoplay=True)
319
  out_text_gr = gr.Text(label="Metrics")
320
+ base_gr = gr.Text(label="Base64")
321
 
322
  tts_button.click(
323
  predict,
 
327
  ref_gr,
328
  normalize_text,
329
  ],
330
+ outputs=[audio_gr, out_text_gr, base_gr],
331
  api_name="predict",
332
  )
333