Yilin0601 commited on
Commit
35bcda1
·
verified ·
1 Parent(s): c7f56a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import pipeline
9
  # --------------------------------------------------
10
  asr = pipeline(
11
  "automatic-speech-recognition",
12
- model="facebook/wav2vec2-large-960h-lv60-self"
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": "tts_models/es/tacotron2-DDC",
49
- "French": "tts_models/fr/tacotron2",
50
  "German": "tts_models/de/tacotron2",
51
- "Chinese": "tts_models/zh/tacotron2", # 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": "tts_models/ja/tacotron2", # Same note
56
  "Italian": "tts_models/it/tacotron2", # Same note
57
- "Korean": "tts_models/ko/tacotron2" # Same note
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
  # --------------------------------------------------