speech-to-speech-translation

Sleeping

peterdamn commited on Jul 18, 2023

Commit

b731530

1 Parent(s): 4c913b4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,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", padding='max_length', truncation=True)
     speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
     return speech.cpu()