import gradio as gr

from transformers import pipeline

english_translator = gr.load("models/arpanghoshal/EmoRoBERTa")


def generate_text(text):
    text = english_translator(text)
    return text


with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            seed = gr.Text(label="Input Phrase")
        with gr.Column():
            text = gr.Text(label="Generated Text")
    btn = gr.Button("Generate")
    btn.click(generate_text, inputs=[seed], outputs=[text])

demo.launch()