Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
pipe = pipeline("translation", model="t5-base") | |
def translate(text, target_language): | |
return pipe(text, target_language=target_language)[0]["translation_text"] | |
with gr.Blocks() as demo: | |
with gr.Row(): | |
with gr.Column(): | |
english = gr.Textbox(label="English text") | |
language_select = gr.Dropdown(label="Translate to", choices=["German", "French", "Lao"]) | |
translate_btn = gr.Button(label="Translate") | |
with gr.Column(): | |
translated = gr.Textbox(label="Translated Text") | |
def translate_callback(): | |
target_language = "de" if language_select.value == "German" else "fr" if language_select.value == "French" else "lo" | |
translated.value = translate(english.value, target_language) | |
translate_btn.onclick(translate_callback) | |
examples = gr.Examples( | |
examples=["Welcome, please select your language.", "Thank you!"], | |
inputs=[english]) | |
demo.launch() | |