Update app.py

app.py

@@ -5,25 +5,27 @@ from huggingface_hub import hf_hub_download
 # Télécharger le modèle depuis Hugging Face
 model_path = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="lid.218.bin")
 
-# Charger le modèle
+# Charger le modèle avec FastText
 model = fasttext.load_model(model_path)
 
 def predict_language(text):
     # Supprimer les retours à la ligne et les espaces inutiles
     text = text.replace('\n', ' ').strip()
     
-    # Effectuer la prédiction
+    # Effectuer la prédiction de la langue
     labels, probs = model.predict(text)
     
-    # Retourner la langue et la probabilité
+    # Retourner la langue prédite et la probabilité correspondante
     return labels[0], probs[0]
 
 # Interface Gradio
-iface = gr.Interface(fn=predict_language, 
-                     inputs=gr.Textbox(label="Entrez votre texte ici"), 
-                     outputs=[gr.Label(), gr.Label()],
-                     live=True)
+iface = gr.Interface(
+    fn=predict_language, 
+    inputs=gr.Textbox(label="Entrez votre texte ici"), 
+    outputs=[gr.Text(label="Langue prédite"), gr.Text(label="Probabilité")],
+    live=True
+)
 
-# Lancer l'application
+# Lancer l'application Gradio
 if __name__ == "__main__":
     iface.launch(share=True)