Spaces:
Sleeping
Sleeping
| 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() | |