vladelesin commited on
Commit
dd057ae
1 Parent(s): fbba0c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import VitsModel, VitsTokenizer, FSMTForConditionalGeneration,
14
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
15
 
16
  # Transform audio to en text
17
- asr_pipe = pipeline("automatic-speech-recognition", model="asapp/sew-d-tiny-100k-ft-ls100h", device=device)
18
 
19
  # Translate en to rus text
20
  translation_en_to_rus = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
 
14
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
15
 
16
  # Transform audio to en text
17
+ asr_pipe = pipeline("automatic-speech-recognition", model="asapp/sew-d-mid-400k-ft-ls100h", device=device)
18
 
19
  # Translate en to rus text
20
  translation_en_to_rus = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")