PablitoGil14's picture
Update app.py
44f0370 verified
raw
history blame contribute delete
902 Bytes
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)