vladelesin commited on
Commit
62a7e0b
·
1 Parent(s): fe3bd76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -26,12 +26,10 @@ translation_en_to_rus = pipeline("translation", model="Helsinki-NLP/opus-mt-en-r
26
  model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
27
  tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
28
 
29
- def transform_audio_to_speech_en(audio):
30
- outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"})
31
- return outputs["text"]
32
 
33
- def translator(text):
34
- translated_text = translation_en_to_rus(text)
 
35
  return translated_text[0]['translation_text']
36
 
37
  def synthesise(text):
 
26
  model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
27
  tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
28
 
 
 
 
29
 
30
+ def translator(audio):
31
+ en_text = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"})
32
+ translated_text = translation_en_to_rus(en_text["text"])
33
  return translated_text[0]['translation_text']
34
 
35
  def synthesise(text):