BissakaAI commited on
Commit
55bbc1c
·
verified ·
1 Parent(s): f26c5be

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
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