Spaces:
Runtime error
Runtime error
Auxiliarytrinket
commited on
Commit
•
ca365d2
1
Parent(s):
814f850
Update app.py
Browse files
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)
|