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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -7,15 +7,16 @@ 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():
11
- target_language = "de" if language_select.value == "German" else "fr" if language_select.value == "French" else "lo"
12
- translated.value = translate(english.value, target_language)
 
13
 
14
  def main():
15
  with gr.Interface(
16
  fn=translate_callback,
17
- inputs=[gr.inputs.Textbox(label="English text")],
18
- outputs=[gr.outputs.Textbox(label="Translated Text")],
19
  title="Text Translation",
20
  description="Translate English text to German, French, or Lao."
21
  ) as iface:
@@ -23,3 +24,4 @@ def main():
23
 
24
  if __name__ == "__main__":
25
  main()
 
 
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:
 
24
 
25
  if __name__ == "__main__":
26
  main()
27
+