VeyVey commited on
Commit
578a906
·
1 Parent(s): e50b5f0

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+ from transformers import pipeline
4
+
5
+ pipe = pipeline("translation", model="t5-base")
6
+
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()