import gradio as gr
import fasttext

# Charger le modèle de FastText pour l'identification des langues
model = fasttext.load_model("lid.176.bin")  # Assure-toi que ce modèle est téléchargé ou accessible

# Fonction de prédiction de la langue
def predict_language(text):
    prediction = model.predict(text)
    return prediction[0][0]  # Retourne la langue prédite

# Interface Gradio
iface = gr.Interface(fn=predict_language, inputs="text", outputs="text", live=True)

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