eljandoubi commited on
Commit
1ca30ca
1 Parent(s): a069f58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def synthesise(text):
29
  with torch.no_grad():
30
  speech = model(inputs["input_ids"].to(device))
31
 
32
- return speech.cpu()
33
 
34
 
35
  def speech_to_speech_translation(audio):
 
29
  with torch.no_grad():
30
  speech = model(inputs["input_ids"].to(device))
31
 
32
+ return speech.audio[0].cpu()
33
 
34
 
35
  def speech_to_speech_translation(audio):