MrFrijo commited on
Commit
868a628
·
verified ·
1 Parent(s): 85eb15f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -1,9 +1,26 @@
1
  import gradio as gr
2
 
3
  def translate_text(text, src_lang, target_lang):
4
- # Ta fonction de traduction ici
 
 
 
 
 
 
 
5
  return translated_text
6
 
7
- interface = gr.Interface(fn=translate_text, inputs=[gr.Textbox(), gr.Radio(["fr", "li"]), gr.Radio(["fr", "li"])], outputs="text")
 
 
 
 
 
 
 
 
 
8
 
 
9
  interface.launch()
 
1
  import gradio as gr
2
 
3
  def translate_text(text, src_lang, target_lang):
4
+ # Préparer les entrées pour le modèle
5
+ tokenized_text = tokenizer(text, return_tensors="pt")
6
+
7
+ # Effectuer la traduction
8
+ translated = model.generate(**tokenized_text)
9
+
10
+ # Convertir et retourner le texte traduit
11
+ translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
12
  return translated_text
13
 
14
+ # Créer l'interface Gradio
15
+ interface = gr.Interface(
16
+ fn=translate_text,
17
+ inputs=[
18
+ gr.Textbox(label="Texte à traduire"),
19
+ gr.Dropdown(choices=["fr", "li"], label="Langue Source"),
20
+ gr.Dropdown(choices=["fr", "li"], label="Langue Cible")
21
+ ],
22
+ outputs="text"
23
+ )
24
 
25
+ # Lancer l'interface
26
  interface.launch()