Update app.py

app.py

@@ -31,15 +31,14 @@ def transform_audio_to_speech_en(audio):
     return outputs["text"]
 
 def translator(text):
-    translated = translator_model.generate(**translator_tokenizer(text, return_tensors="pt", padding=True))
-    translated_text = translator_tokenizer.decode(translated[0], skip_special_tokens=True)
-    return translated_text
+  translated_text = translation_en_to_rus(text)
+  return translated_text[0]['translation_text']
 
 def synthesise(translated_text):
     translated_text = translator(translated_text)
-    inputs = vits_tokenizer(translated_text, return_tensors="pt")
+    inputs = tokenizer(translated_text, return_tensors="pt")
     with torch.no_grad():
-        speech = vits_model(**inputs).waveform
+        speech = model(**inputs).waveform
     return speech.cpu()
 
 def speech_to_speech_translation(audio):