File size: 902 Bytes
c280be0 44f0370 c280be0 405414d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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)
|