import gradio as gr

from transformers import pipeline

translator = pipeline("translation", model="vsrinivas/marian-finetuned-kde4-en-to-hi")

def translate(input):
    output = translator(input)
    return output['translation_text']

demo = gr.Interface(fn=translate,
                    inputs=[gr.Textbox(label="English text that you want to translate into Hindi !", lines=6)],
                    outputs=[gr.Textbox(label="Translated Hindi Text !!!", lines=3)],
                    title="English to Hindi Text translation test and demo app by Srinivas.V ..",
                    description="Translates English text into Hindi text",
                    allow_flagging="never"
                   )
demo.launch()