import gradio as gr
from transformers import pipeline

pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")

def predict(text):
  return pipe(text)[0]["translation_text"]

with gr.Blocks() as demo:
    english_inp = gr.Textbox(label="English Sentence")
    spanish_op = gr.Textbox(label="Spanish Translation")
    translate_btn = gr.Button("Translate")
    translate_btn.click(fn=predict, inputs=english_inp, outputs=spanish_op)

demo.launch()