import gradio as gr
from transformers import pipeline

# repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
repo_id = "PablitoGil14/clasificador-sentiment"

classifier = pipeline("text-classification", model="PablitoGil14/clasificador-sentiment")

# Definimos una función que se encarga de llevar a cabo las predicciones
def predict(str):
    dic_label_score = classifier(str)[0]
    label = dic_label_score['label']
    score = dic_label_score['score']
    dicc = {'LABEL_0': 'Positive', 'LABEL_1': 'Neutral', 'LABEL_2': 'Negative'}
    label = dicc[label]
    return label # + ', con score ' + str(score))  
    
# Creamos la interfaz y la lanzamos. 
gr.Interface(fn=predict, inputs="textbox", outputs="textbox", examples=['Le cose della scuola che mi fanno sentire stressato ma non troppo sono le verifiche, interrogazioni, compiti.','Che insegnante meravigliosa e come spiega bene.']).launch(share=True)