yuvscherbatov
commited on
Commit
•
a824dd6
1
Parent(s):
4fbf54b
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
19 |
asr_pipe = pipeline("automatic-speech-recognition", model="asapp/sew-d-tiny-100k-ft-ls100h", device=device)
|
20 |
|
21 |
#eng text to rus text translation
|
22 |
-
translation_pipe = pipeline("translation", model="
|
23 |
|
24 |
#rus text to rus speech transformation
|
25 |
vits_model = VitsModel.from_pretrained("facebook/mms-tts-rus")
|
|
|
19 |
asr_pipe = pipeline("automatic-speech-recognition", model="asapp/sew-d-tiny-100k-ft-ls100h", device=device)
|
20 |
|
21 |
#eng text to rus text translation
|
22 |
+
translation_pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
|
23 |
|
24 |
#rus text to rus speech transformation
|
25 |
vits_model = VitsModel.from_pretrained("facebook/mms-tts-rus")
|