Auxiliarytrinket commited on
Commit
ca365d2
1 Parent(s): 814f850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -10,14 +10,13 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
10
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=device) #Тут добавил tiny, потому что модель станет более компактной
11
  translater = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru") # Инициализация модели для перевода текста на русский язык
12
 
 
 
 
13
  def translate(audio, translater: pipeline = translater): # Определение функции для перевода аудио в текст
14
  outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"}) # Получение текстового представления аудио
15
  return translater(outputs['text'])[0]['translation_text'] # Возврат переведенного текста
16
 
17
-
18
- model = VitsModel.from_pretrained("facebook/mms-tts-rus") # Загрузка модели для генерации речи на русском языке
19
- tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-rus") # Загрузка токенизатора для модели
20
-
21
  def synthesise(text: str, tokenizer: AutoTokenizer = tokenizer, model: VitsModel = model): # Определение функции для синтеза речи из текста
22
  inputs = tokenizer(text, return_tensors="pt") # Создание токенизированного представления текста
23
  # print(inputs)
 
10
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=device) #Тут добавил tiny, потому что модель станет более компактной
11
  translater = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru") # Инициализация модели для перевода текста на русский язык
12
 
13
+ model = VitsModel.from_pretrained("facebook/mms-tts-rus") # Загрузка модели для генерации речи на русском языке
14
+ tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-rus") # Загрузка токенизатора для модели
15
+
16
  def translate(audio, translater: pipeline = translater): # Определение функции для перевода аудио в текст
17
  outputs = asr_pipe(audio, max_new_tokens=256, generate_kwargs={"task": "translate"}) # Получение текстового представления аудио
18
  return translater(outputs['text'])[0]['translation_text'] # Возврат переведенного текста
19
 
 
 
 
 
20
  def synthesise(text: str, tokenizer: AutoTokenizer = tokenizer, model: VitsModel = model): # Определение функции для синтеза речи из текста
21
  inputs = tokenizer(text, return_tensors="pt") # Создание токенизированного представления текста
22
  # print(inputs)