Update model.py
Browse files
model.py
CHANGED
|
@@ -37,7 +37,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 37 |
# Load ASR
|
| 38 |
# --------------------------
|
| 39 |
ASR_MODEL = "facebook/seamless-m4t-v2-large"
|
| 40 |
-
processor = AutoProcessor.from_pretrained(ASR_MODEL, token=HF_TOKEN)
|
| 41 |
asr_model = SeamlessM4Tv2ForSpeechToText.from_pretrained(ASR_MODEL, token=HF_TOKEN, use_fast=False).to(device)
|
| 42 |
asr_model.eval()
|
| 43 |
|
|
@@ -51,8 +51,8 @@ for lang, tts_name in {
|
|
| 51 |
# "hausa": "facebook/mms-tts-hau",
|
| 52 |
}.items():
|
| 53 |
print(f"Loading TTS model for {lang}...")
|
| 54 |
-
tts_proc = AutoProcessor.from_pretrained(tts_name, token=HF_TOKEN)
|
| 55 |
-
tts_mod = VitsModel.from_pretrained(tts_name, token=HF_TOKEN).to(device)
|
| 56 |
tts_mod.eval()
|
| 57 |
tts_models[lang] = {"processor": tts_proc, "model": tts_mod}
|
| 58 |
|
|
|
|
| 37 |
# Load ASR
|
| 38 |
# --------------------------
|
| 39 |
ASR_MODEL = "facebook/seamless-m4t-v2-large"
|
| 40 |
+
processor = AutoProcessor.from_pretrained(ASR_MODEL, token=HF_TOKEN,use_fast=False)
|
| 41 |
asr_model = SeamlessM4Tv2ForSpeechToText.from_pretrained(ASR_MODEL, token=HF_TOKEN, use_fast=False).to(device)
|
| 42 |
asr_model.eval()
|
| 43 |
|
|
|
|
| 51 |
# "hausa": "facebook/mms-tts-hau",
|
| 52 |
}.items():
|
| 53 |
print(f"Loading TTS model for {lang}...")
|
| 54 |
+
tts_proc = AutoProcessor.from_pretrained(tts_name, token=HF_TOKEN,use_fast=False)
|
| 55 |
+
tts_mod = VitsModel.from_pretrained(tts_name, token=HF_TOKEN,use_fast=False).to(device)
|
| 56 |
tts_mod.eval()
|
| 57 |
tts_models[lang] = {"processor": tts_proc, "model": tts_mod}
|
| 58 |
|