Update app.py

app.py

@@ -34,8 +34,11 @@ def translate(audio):
   return translated_text[0]['translation_text']
 
 def synthesise(text):
-    speech = model.generate(**tokenizer(text, return_tensors="pt"))
-    return speech.to("cpu")
+    translated_text = translator(text)
+    inputs = tokenizer(translated_text, return_tensors="pt")
+    with torch.no_grad():
+        speech = model(**inputs).waveform
+    return speech.cpu()
 
 def speech_to_speech_translation(audio):
     translated_text = translate(audio)