Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import NllbTokenizer, AutoModelForSeq2SeqLM
 import torch
 
 MODEL_NAME = "Tamazight-NLP/NLLB-200-600M-Tamazight-All-Data-3-epoch"
@@ -28,7 +28,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+tokenizer = NllbTokenizer.from_pretrained(MODEL_NAME)
 
 
 def translate(text, source_lang, target_lang, max_length=238, num_beams=4):