speech-to-speech-translation

Runtime error

fisheggg commited on Aug 28, 2023

Commit

f5ef656

•

1 Parent(s): b69950b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,8 +29,8 @@ def translate(audio):
 def synthesise(text):
-    inputs = processor(text=text, return_tensors="pt")[:, :600]
-    speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
     return speech.cpu()

 def synthesise(text):
+    inputs = processor(text=text, return_tensors="pt")
+    speech = model.generate_speech(inputs["input_ids"][:, :600].to(device), speaker_embeddings.to(device), vocoder=vocoder)
     return speech.cpu()