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):
    result = classifier(sentence)[0]
    label,score = result['label'],result['score']
    if label == 'Sí':
        result = {'Sí':score,'No':1-score}
    else:
        result = {'No':score,'Sí':1-score}
    return result


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()