maj model

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 # --------------------------------------------------
 # Chargement du pipeline NLLB
 # --------------------------------------------------
-MODEL_NAME = "facebook/nllb-200-3.3B"
+MODEL_NAME = "facebook/nllb-200-distilled-1.3B"
 
 device = 0 if torch.cuda.is_available() else -1
 print(f"🚀 Chargement du modèle {MODEL_NAME} sur {'GPU' if device == 0 else 'CPU'}...")
@@ -13,7 +13,9 @@ print(f"🚀 Chargement du modèle {MODEL_NAME} sur {'GPU' if device == 0 else '
 translator = pipeline(
     "translation",
     model=MODEL_NAME,
-    device=device
+    device=device,
+    src_lang="fra_Latn",
+    tgt_lang="ewe_Latn"
 )
 
 # --------------------------------------------------
@@ -87,4 +89,4 @@ with gr.Blocks(title="🌍 Traduction EWE") as demo:
         """
     )
 
-demo.launch()
+demo.launch()