Annvasileva commited on
Commit
bfddd1a
1 Parent(s): 8391d53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from transformers import pipeline, Wav2Vec2ForCTC, Wav2Vec2Processor, MarianMTModel, MarianTokenizer, T5ForConditionalGeneration, T5Tokenizer, AutoProcessor, AutoModel
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 для распознавания речи на английском.Затем переводит текст на русский с помощью модели t5-small, и осуществляет синтез речи на русском на основе модели Helsinki-NLP/opus-mt-en-ru"
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