Annvasileva
commited on
Commit
•
bfddd1a
1
Parent(s):
8391d53
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
-
from transformers import pipeline, Wav2Vec2ForCTC, Wav2Vec2Processor,
|
4 |
|
5 |
|
6 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
@@ -39,7 +39,7 @@ def speech_to_speech_translation(audio):
|
|
39 |
return russian_speech.numpy()
|
40 |
|
41 |
title = "Speech-to-Speech Translation"
|
42 |
-
description = "Код сначала использует модель facebook/wav2vec2-base-960h для распознавания речи на английском.Затем переводит текст на русский с помощью модели
|
43 |
|
44 |
|
45 |
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
+
from transformers import pipeline, Wav2Vec2ForCTC, Wav2Vec2Processor, AutoProcessor, AutoModel
|
4 |
|
5 |
|
6 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
39 |
return russian_speech.numpy()
|
40 |
|
41 |
title = "Speech-to-Speech Translation"
|
42 |
+
description = "Код сначала использует модель facebook/wav2vec2-base-960h для распознавания речи на английском.Затем переводит текст на русский с помощью модели Helsinki-NLP/opus-mt-en-ru, и осуществляет синтез речи на русском на основе модели suno/bark"
|
43 |
|
44 |
|
45 |
|