Sukumar2005's picture
Update app.py
1489797 verified
from transformers import pipeline
import gradio as gr
# Load translation pipelines once
de_translator = pipeline("translation_en_to_de", model="Helsinki-NLP/opus-mt-en-de")
hi_translator = pipeline("translation_en_to_hi", model="Helsinki-NLP/opus-mt-en-hi")
def translate_text(text):
# Translate English β†’ German
result_de = de_translator(text, max_length=40)[0]['translation_text']
# Translate English β†’ Hindi
result_hi = hi_translator(text, max_length=40)[0]['translation_text']
return result_de, result_hi
# Gradio interface
demo = gr.Interface(
fn=translate_text,
inputs=gr.Textbox(lines=2, placeholder="Enter English text here..."),
outputs=[
gr.Textbox(label="Helsinki-NLP Translation (English β†’ German)"),
gr.Textbox(label="Helsinki-NLP Translation (English β†’ Hindi)")
],
title="πŸ“ Text Translator",
description="Enter English text and get translations into German and Hindi using public Helsinki-NLP models"
)
if __name__ == "__main__":
demo.launch()