app.py

import gradio as gr
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

# Charger le modèle depuis Hugging Face
model_name = "MaroneAI/Niani-nllb-Wolof-To-Frensh-615M"

# ✅ Correction : ignorer le tokenizer.json et forcr 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)

demo = gr.Interface(
    fn=translate,
    inputs=gr.Textbox(lines=3, placeholder="Entrez un texte en Wolof..."),
    outputs=gr.Textbox(label="Traduction en wolof"),
    title="Niani Translator-2 🍉",
    description="Modèle de traduction Wolof → Français fine-tuné par MaroneAI. 🚧---(Il préférable de traduire avec des mots wolof valides. Sinon ça peut beaucoup moins marcher. Essayer de traduire des mots wolof qui n'existent pas mais que vous croyez exister par le son est une trés grande ERREUR!" 
)

if __name__ == "__main__":
    demo.launch()