Update app.py

app.py

@@ -1,5 +1,4 @@
-# -*- coding: utf-8 -*-
-"""app.ipynb
+"""
 Automatically generated by Colaboratory.
 Original file is located at
     https://colab.research.google.com/drive/16MxXQeF3O0htL9eQ61aa6ZxnApGg9TKN
@@ -9,17 +8,18 @@ import gradio as gr
 import numpy as np
 import torch
 
-from transformers import pipeline, VitsModel, VitsTokenizer, FSMTForConditionalGeneration, FSMTTokenizer
+from transformers import pipeline
+from transformers import VitsModel, VitsTokenizer, FSMTForConditionalGeneration, FSMTTokenizer
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-#eng audio to text transformation
+# Transform audio to en text
 asr_pipe = pipeline("automatic-speech-recognition", model="asapp/sew-d-tiny-100k-ft-ls100h", device=device)
 
-#eng text to rus text translation
-translation_pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
+# Translate en to rus text
+translation_en_to_rus = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
 
-#rus text to rus speech transformation
+# Create speech from rus text
 vits_model = VitsModel.from_pretrained("facebook/mms-tts-rus")
 vits_tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-rus")
 
@@ -28,7 +28,7 @@ def transform_audio_to_speech_en(audio):
     return outputs["text"]
 
 def translator(text):
-  translated_text = translation_pipe(text)
+  translated_text = translation_en_to_rus(text)
   return translated_text[0]['translation_text']
 
 def synthesise(translated_text):