Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,14 @@ classifier = pipeline('text-classification', model='joheras/clasificador-ser-est
|
|
4 |
|
5 |
|
6 |
def classify(sentence):
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|