fasttext / app.py
samdo20's picture
Update app.py
b2c3872 verified
raw
history blame
523 Bytes
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()