Update app.py

app.py

@@ -1,12 +1,26 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-tokenizer = AutoTokenizer.from_pretrained("MaroneAI/Niani-Translator")
-model = AutoModelForSeq2SeqLM.from_pretrained("MaroneAI/Niani-Translator")
+# Charger le modèle depuis ton repo Hugging Face
+model_name = "MaroneAI/Niani-Translator"
 
-def translate_fr_wo(text):
-    inputs = tokenizer(text, return_tensors="pt")
-    outputs = model.generate(**inputs, max_new_tokens=128)
+# ✅ Correction : ignorer le tokenizer.json et forcer SentencePiece
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+def translate(text):
+    inputs = tokenizer(text, return_tensors="pt", padding=True)
+    outputs = model.generate(**inputs, max_length=256)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-gr.Interface(fn=translate_fr_wo, inputs="text", outputs="text", title="Niani-Translator").launch()
+demo = gr.Interface(
+    fn=translate,
+    inputs=gr.Textbox(lines=3, placeholder="Entrez un texte en français..."),
+    outputs=gr.Textbox(label="Traduction en wolof"),
+    title="Niani Translator 🌍",
+    description="Modèle de traduction Français → Wolof fine-tuné par MaroneAI."
+)
+
+if __name__ == "__main__":
+    demo.launch()
+