entohi / app.py
vijaydhanda's picture
Updated
3a4b804
raw
history blame
867 Bytes
import gradio as gr
from transformers import pipeline
model = pipeline("translation", model="AbhirupGhosh/opus-mt-finetuned-en-hi")
def translate(text):
translations = model(text)
output = translations[0]['translation_text']
return output
gr.Interface(
fn=translate,
inputs = gr.Textbox(label="EN"),
outputs = gr.Textbox(label="HI"),
title="Translate English to Hindi",
description="Translation using Helsinki-NLP Opus MT model.",
examples= [
"Traduire une langue est un problème très complexe et pourtant ce modèle de Helsinki-NLP y arrive à la fois plutôt bien et rapidement, tu en conviens?",
"J'aurais préféré qu'ils soient tous présent lors de la démonstration plutôt que d'avoir à me répéter pour chacun d'entre-eux.",
],
allow_flagging="never"
).launch(debug=True, enable_queue=True)