joheras commited on
Commit
e7ecab1
verified
1 Parent(s): 4c80fe5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -4,7 +4,14 @@ classifier = pipeline('text-classification', model='joheras/clasificador-ser-est
4
 
5
 
6
  def classify(sentence):
7
- return classifier(sentence)[0]
 
 
 
 
 
 
 
8
 
9
  demo = gr.Interface(fn=classify, inputs="text", outputs="label", examples=["Ellas compraron un saco de patatas", "Ellas eran jugando con sus amigas", "Ellas estaban jugando con sus amigas"])
10
  demo.launch()
 
4
 
5
 
6
  def classify(sentence):
7
+ result = classifier(sentence)[0]
8
+ label,score = result['label'],result['score']
9
+ if label == 'S铆':
10
+ result = {'S铆':score,'No':1-score}
11
+ else:
12
+ result = {'No':score,'S铆':1-score}
13
+ return result
14
+
15
 
16
  demo = gr.Interface(fn=classify, inputs="text", outputs="label", examples=["Ellas compraron un saco de patatas", "Ellas eran jugando con sus amigas", "Ellas estaban jugando con sus amigas"])
17
  demo.launch()