import gradio as gr
from transformers import pipeline
classifier = pipeline('text-classification', model='joheras/clasificador-ser-estar-cambiadas-con-error-bert-base-v2')


def classify(sentence):
    return classifier(sentence)[0]

demo = gr.Interface(fn=classify, inputs="text", outputs="label", examples=["Ellas compraron un saco de patatas", "Ellas eran jugando con sus amigas", "Ellas estaban jugando con sus amigas"])
demo.launch()