Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from transformers import pipeline
|
|
9 |
# --------------------------------------------------
|
10 |
asr = pipeline(
|
11 |
"automatic-speech-recognition",
|
12 |
-
model="facebook/wav2vec2-
|
13 |
)
|
14 |
|
15 |
# --------------------------------------------------
|
@@ -45,16 +45,16 @@ translation_tasks = {
|
|
45 |
|
46 |
# TTS models (some may not exist or may be unofficial)
|
47 |
tts_models = {
|
48 |
-
"Spanish": "
|
49 |
-
"French": "
|
50 |
"German": "tts_models/de/tacotron2",
|
51 |
-
"Chinese": "
|
52 |
"Russian": "tts_models/ru/tacotron2", # Same note
|
53 |
"Arabic": "tts_models/ar/tacotron2", # Same note
|
54 |
"Portuguese": "tts_models/pt/tacotron2", # Same note
|
55 |
-
"Japanese": "
|
56 |
"Italian": "tts_models/it/tacotron2", # Same note
|
57 |
-
"Korean": "
|
58 |
}
|
59 |
|
60 |
# --------------------------------------------------
|
|
|
9 |
# --------------------------------------------------
|
10 |
asr = pipeline(
|
11 |
"automatic-speech-recognition",
|
12 |
+
model="facebook/wav2vec2-base-960h"
|
13 |
)
|
14 |
|
15 |
# --------------------------------------------------
|
|
|
45 |
|
46 |
# TTS models (some may not exist or may be unofficial)
|
47 |
tts_models = {
|
48 |
+
"Spanish": "myshell-ai/MeloTTS-Spanish",
|
49 |
+
"French": "myshell-ai/MeloTTS-French",
|
50 |
"German": "tts_models/de/tacotron2",
|
51 |
+
"Chinese": "myshell-ai/MeloTTS-English-v2", # Verify if this actually exists on Hugging Face
|
52 |
"Russian": "tts_models/ru/tacotron2", # Same note
|
53 |
"Arabic": "tts_models/ar/tacotron2", # Same note
|
54 |
"Portuguese": "tts_models/pt/tacotron2", # Same note
|
55 |
+
"Japanese": "myshell-ai/MeloTTS-Japanese", # Same note
|
56 |
"Italian": "tts_models/it/tacotron2", # Same note
|
57 |
+
"Korean": "myshell-ai/MeloTTS-Korean" # Same note
|
58 |
}
|
59 |
|
60 |
# --------------------------------------------------
|