Update app.py

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+from transformers import MarianMTModel, MarianTokenizer
+
+MODEL_NAME = "mafromedia/yaaba-fr-mo-model"
+
+tokenizer = MarianTokenizer.from_pretrained(MODEL_NAME)
+model = MarianMTModel.from_pretrained(MODEL_NAME)
+
+def translate(text, direction):
+    if not text.strip():
+        return "❗ Please enter text."
+    inputs = tokenizer(text, return_tensors="pt", padding=True)
+    outputs = model.generate(**inputs, max_length=64)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+with gr.Blocks(title="Yaaba AI – French ↔ Mooré Translator") as demo:
+    gr.Markdown("# 🌍 Yaaba AI – French ↔ Mooré Translator")
+    with gr.Row():
+        txt = gr.Textbox(label="Enter text", placeholder="Bonjour")
+        direction = gr.Radio(["French → Mooré", "Mooré → French"], value="French → Mooré")
+    out = gr.Textbox(label="Translation")
+    btn = gr.Button("Translate")
+    btn.click(fn=translate, inputs=[txt, direction], outputs=out)
+    gr.Examples(
+        [["Bonjour", "French → Mooré"],
+         ["Laafi beme?", "Mooré → French"],
+         ["Bienvenue", "French → Mooré"]],
+        inputs=[txt, direction]
+    )
+
+demo.launch()