VeyVey commited on
Commit
f910e58
·
1 Parent(s): 2bd3048

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -18
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import gradio as gr
2
-
3
  from transformers import pipeline
4
 
5
  pipe = pipeline("translation", model="t5-base")
@@ -7,23 +6,20 @@ 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
- with gr.Blocks() as demo:
11
- with gr.Row():
12
- with gr.Column():
13
- english = gr.Textbox(label="English text")
14
- language_select = gr.Dropdown(label="Translate to", choices=["German", "French", "Lao"])
15
- translate_btn = gr.Button(label="Translate")
16
- with gr.Column():
17
- translated = gr.Textbox(label="Translated Text")
18
-
19
- def translate_callback():
20
- target_language = "de" if language_select.value == "German" else "fr" if language_select.value == "French" else "lo"
21
- translated.value = translate(english.value, target_language)
22
 
23
- translate_btn.onclick(translate_callback)
 
 
 
 
 
 
 
 
24
 
25
- examples = gr.Examples(
26
- examples=["Welcome, please select your language.", "Thank you!"],
27
- inputs=[english])
28
 
29
- demo.launch()
 
1
  import gradio as gr
 
2
  from transformers import pipeline
3
 
4
  pipe = pipeline("translation", model="t5-base")
 
6
  def translate(text, target_language):
7
  return pipe(text, target_language=target_language)[0]["translation_text"]
8
 
9
+ def translate_callback():
10
+ target_language = "de" if language_select.value == "German" else "fr" if language_select.value == "French" else "lo"
11
+ translated.value = translate(english.value, target_language)
 
 
 
 
 
 
 
 
 
12
 
13
+ def main():
14
+ with gr.Interface(
15
+ fn=translate_callback,
16
+ inputs=[gr.inputs.Textbox(label="English text")],
17
+ outputs=[gr.outputs.Textbox(label="Translated Text")],
18
+ title="Text Translation",
19
+ description="Translate English text to German, French, or Lao."
20
+ ) as iface:
21
+ iface.launch()
22
 
23
+ if __name__ == "__main__":
24
+ main()
 
25