PablitoGil14 commited on
Commit
c280be0
verified
1 Parent(s): d547c7c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
5
+ repo_id = "PablitoGil14/clasificador-sentiment"
6
+
7
+ classifier = pipeline("text-classification", model="PablitoGil14/clasificador-sentiment")
8
+
9
+ # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
10
+ def predict(str):
11
+ dic_label_score = classifier(str)[0]
12
+ label = dic_label_score['label']
13
+ score = dic_label_score['score']
14
+ dicc = {'LABEL_0': 'Positive', 'LABEL_1': 'Neutral', 'LABEL_2': 'Negative'}
15
+ label = dicc[label]
16
+ return label # + ', con score ' + str(score))
17
+
18
+ # Creamos la interfaz y la lanzamos.
19
+ gr.Interface(fn=predict, inputs="textbox", outputs="textbox", examples=['Ho fatto un esame pessimo, non mi piace studiare.','Che insegnante meravigliosa e come spiega bene.']).launch(share=True)