Bgk Injector SqLi commited on
Commit
9b5d5f1
·
1 Parent(s): d0b5038

Fix: Use NllbTokenizer instead of AutoTokenizer

Browse files

- Fixes AttributeError with NLLB model loading
- AutoTokenizer can't auto-detect NLLB tokenizer class
- Now using NllbTokenizer explicitly

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,9 +81,9 @@ def load_models():
81
  print("✅ TTS prêt!")
82
 
83
  # Translation (Dioula → Français)
84
- from transformers import AutoModelForSeq2SeqLM
85
  print("⏳ Chargement du modèle de traduction (Dioula → Français)...")
86
- translate_tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
87
  translate_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M").to(device)
88
  print("✅ Traduction prête!")
89
 
 
81
  print("✅ TTS prêt!")
82
 
83
  # Translation (Dioula → Français)
84
+ from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
85
  print("⏳ Chargement du modèle de traduction (Dioula → Français)...")
86
+ translate_tokenizer = NllbTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
87
  translate_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M").to(device)
88
  print("✅ Traduction prête!")
89