Commit
·
62a7e0b
1
Parent(s):
fe3bd76
Update app.py
Browse files
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(
|
34 |
-
|
|
|
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):
|