Spaces:
Running
Running
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
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 =
|
| 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 |
|