File size: 946 Bytes
7a680a1 ec81572 a270abe be9881b e3f75fd a270abe be9881b e67adcc be9881b b2c3872 a270abe e67adcc a270abe e67adcc a270abe b2c3872 e67adcc b2c3872 e67adcc b2c3872 a270abe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import fasttext
import gradio as gr
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 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 de la langue
labels, probs = model.predict(text)
# 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.Text(label="Langue prédite"), gr.Text(label="Probabilité")],
live=True
)
# Lancer l'application Gradio
if __name__ == "__main__":
iface.launch(share=True)
|