speech-to-speech-translation

Runtime error

fisheggg commited on Aug 28, 2023

Commit

b69950b

•

1 Parent(s): 69f1cf5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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