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()
|