File size: 623 Bytes
e3f75fd 7a680a1 ec81572 be9881b e3f75fd be9881b e3f75fd be9881b b2c3872 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from huggingface_hub import hf_hub_download
import fasttext
import gradio as gr
# Télécharger le modèle depuis Hugging Face
model_path = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="model.bin")
# Charger le modèle avec fastText
model = fasttext.load_model(model_path)
# 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()
|