File size: 1,037 Bytes
24cfc7d
675da53
 
 
1489797
 
675da53
 
1489797
 
 
 
 
675da53
1489797
675da53
 
 
 
1489797
 
675da53
 
1489797
675da53
24cfc7d
 
675da53
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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()