13hbeltran's picture
Update app.py
38eeae5 verified
import gradio as gr
from transformers import pipeline
models = {
"Spanish": pipeline("translation", model="Helsinki-NLP/opus-mt-en-es"),
"German": pipeline("translation", model="Helsinki-NLP/opus-mt-en-de"),
"Japanese": pipeline("translation", model="staka/fugumt-en-ja"),
"Ukrainian": pipeline("translation", model="Helsinki-NLP/opus-mt-en-uk"),
"Russian": pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru"),
}
def translate(text, language):
if not text:
return "Please enter an English sentence."
return models[language](text)[0]["translation_text"]
with gr.Blocks(title="Short Translation") as demo:
gr.Markdown("# Short Translation")
with gr.Row():
inp = gr.Textbox(label="English Sentence", lines=2, placeholder="Type your English sentence here...")
lang = gr.Radio(["Spanish", "German", "Japanese", "Ukrainian", "Russian"], value="Spanish", label="Translation Language")
out = gr.Textbox(label="Translation", interactive=False)
btn = gr.Button("Translate")
btn.click(translate, inputs=[inp, lang], outputs=out)
demo.launch()