emoroberta / app.py
r-o-y-a's picture
Update app.py
63a4784
raw
history blame
No virus
509 Bytes
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()