# gradio demo
import gradio as gr
from transformers import pipeline

pipe = pipeline("sentiment-analysis")

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

iface = gr.Interface(
  fn=predict,
  inputs='text',
  outputs='text',
  examples=[["The movie was amazing"]]
)

iface.launch()