proz commited on
Commit
cb4c63f
·
verified ·
1 Parent(s): 284104a

Update download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +3 -3
download_model.py CHANGED
@@ -1,9 +1,9 @@
1
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
2
 
3
- # --- CHANGEMENT ICI : On pointe vers la V2 ---
4
- MODEL_ID = "Cnam-LMSSC/wav2vec2-french-phonemizer-v2"
5
 
6
  print(f"⬇️ Téléchargement du modèle {MODEL_ID}...")
7
  processor = Wav2Vec2Processor.from_pretrained(MODEL_ID)
8
  model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
9
- print("✅ Modèle téléchargé et mis en cache.")
 
1
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
2
 
3
+ # ON REVIENT À LA VERSION 1 (La valeur sûre)
4
+ MODEL_ID = "Cnam-LMSSC/wav2vec2-french-phonemizer"
5
 
6
  print(f"⬇️ Téléchargement du modèle {MODEL_ID}...")
7
  processor = Wav2Vec2Processor.from_pretrained(MODEL_ID)
8
  model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
9
+ print("✅ Modèle V1 téléchargé.")