Update app.py

app.py

@@ -1,9 +1,15 @@
+"""ML_task3.ipynb
+Automatically generated by Colaboratory.
+Original file is located at
+    https://colab.research.google.com/drive/1DfK6fjkAd9RjVx3MUGfDtAOulvEenk0E?authuser=0#scrollTo=x0BqFNWue3V8
+"""
+
 import gradio as gr
 import numpy as np
 import torch
 from datasets import load_dataset
 
-from transformers import SpeechT5ForTextToSpeech, SpeechT5HifiGan, SpeechT5Processor, pipeline, WhisperProcessor
+from transformers import SpeechT5ForTextToSpeech, SpeechT5HifiGan, SpeechT5Processor, pipeline, WhisperProcessor, VitsModel, VitsTokenizer
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
@@ -16,7 +22,6 @@ processor = WhisperProcessor.from_pretrained(
 translator_en = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
 translator_ru = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
 
-from transformers import VitsModel, VitsTokenizer
 
 model = VitsModel.from_pretrained("facebook/mms-tts-rus")
 tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-rus")