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()