File size: 748 Bytes
85eb15f 868a628 85eb15f 868a628 85eb15f 868a628 85eb15f |
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 |
import gradio as gr
def translate_text(text, src_lang, target_lang):
# Préparer les entrées pour le modèle
tokenized_text = tokenizer(text, return_tensors="pt")
# Effectuer la traduction
translated = model.generate(**tokenized_text)
# Convertir et retourner le texte traduit
translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
return translated_text
# Créer l'interface Gradio
interface = gr.Interface(
fn=translate_text,
inputs=[
gr.Textbox(label="Texte à traduire"),
gr.Dropdown(choices=["fr", "li"], label="Langue Source"),
gr.Dropdown(choices=["fr", "li"], label="Langue Cible")
],
outputs="text"
)
# Lancer l'interface
interface.launch()
|