VeyVey commited on
Commit
70fa9fb
·
1 Parent(s): 57735bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -10
app.py CHANGED
@@ -1,4 +1,3 @@
1
-
2
  import gradio as gr
3
  from transformers import pipeline
4
 
@@ -7,21 +6,42 @@ pipe = pipeline("translation", model="t5-base")
7
  def translate(text, target_language):
8
  return pipe(text, target_language=target_language)[0]["translation_text"]
9
 
10
- def translate_callback(english, language_select, translated):
11
  target_language = "de" if language_select == "German" else "fr" if language_select == "French" else "lo"
12
- translated = translate(english, target_language)
13
- return translated
 
 
 
14
 
15
  def main():
16
- with gr.Interface(
 
 
 
 
17
  fn=translate_callback,
18
- inputs=[gr.inputs.Textbox(label="English text"), "text", gr.inputs.Dropdown(label="Translate to", choices=["German", "French", "Lao"])],
19
- outputs=gr.outputs.Textbox(label="Translated Text"),
20
  title="Text Translation",
21
  description="Translate English text to German, French, or Lao."
22
- ) as iface:
23
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
 
25
  if __name__ == "__main__":
26
  main()
27
-
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
6
  def translate(text, target_language):
7
  return pipe(text, target_language=target_language)[0]["translation_text"]
8
 
9
+ def translate_callback(english, language_select):
10
  target_language = "de" if language_select == "German" else "fr" if language_select == "French" else "lo"
11
+ translated_text = translate(english, target_language)
12
+ return translated_text
13
+
14
+ def clear_callback(english):
15
+ return ""
16
 
17
  def main():
18
+ english_textbox = gr.inputs.Textbox(label="English text")
19
+ language_select = gr.inputs.Dropdown(label="Translate to", choices=["German", "French", "Lao"])
20
+ translated_textbox = gr.outputs.Textbox(label="Translated Text")
21
+
22
+ translate_interface = gr.Interface(
23
  fn=translate_callback,
24
+ inputs=[english_textbox, language_select],
25
+ outputs=translated_textbox,
26
  title="Text Translation",
27
  description="Translate English text to German, French, or Lao."
28
+ )
29
+
30
+ clear_interface = gr.Interface(
31
+ fn=clear_callback,
32
+ inputs=english_textbox,
33
+ outputs=translated_textbox,
34
+ title="Text Translation",
35
+ description="Translate English text to German, French, or Lao."
36
+ )
37
+
38
+ iface = gr.Interface(
39
+ [translate_interface, clear_interface],
40
+ gr.Interface.Column([english_textbox, language_select, translate_interface, clear_interface], labels=["English text", "Translate to", "Translate", "Clear"]),
41
+ gr.Interface.Row([translated_textbox])
42
+ )
43
+
44
+ iface.launch()
45
 
46
  if __name__ == "__main__":
47
  main()