Create app.py

app.py

@@ -0,0 +1,43 @@
+import gradio as gr
+from transformers import MarianMTModel, MarianTokenizer
+
+# Modèle de traduction : anglais → français par défaut
+model_name = "Helsinki-NLP/opus-mt-en-fr"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
+
+def translate(text, target_lang):
+    # Choisir le modèle selon la langue cible
+    if target_lang == "fr":
+        model_name = "Helsinki-NLP/opus-mt-en-fr"
+    elif target_lang == "en":
+        model_name = "Helsinki-NLP/opus-mt-fr-en"
+    else:
+        return "Langue non supportée pour l'instant"
+
+    # Recharger modèle si changé
+    global tokenizer, model
+    if model_name != model.name_or_path:
+        tokenizer = MarianTokenizer.from_pretrained(model_name)
+        model = MarianMTModel.from_pretrained(model_name)
+        model.name_or_path = model_name
+
+    # Traduction
+    batch = tokenizer([text], return_tensors="pt", padding=True)
+    gen = model.generate(**batch)
+    translated = tokenizer.batch_decode(gen, skip_special_tokens=True)[0]
+    return translated
+
+# Interface Gradio
+iface = gr.Interface(
+    fn=translate,
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Tapez le texte ici..."),
+        gr.Dropdown(["fr", "en"], label="Langue cible")
+    ],
+    outputs="text",
+    title="MyTranslator",
+    description="Traduction simple avec détection de langue."
+)
+
+iface.launch()