|
import gradio as gr |
|
from transformers import pipeline |
|
|
|
|
|
repo_id = "PablitoGil14/clasificador-sentiment" |
|
|
|
classifier = pipeline("text-classification", model="PablitoGil14/clasificador-sentiment") |
|
|
|
|
|
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 |
|
|
|
|
|
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) |
|
|