Kaushik Bar
commited on
Commit
·
8ac7f12
1
Parent(s):
70ebd67
first commit
Browse files
app.py
CHANGED
|
@@ -114,7 +114,7 @@ def sequence_to_classify(text, labels):
|
|
| 114 |
probs = probs.softmax(dim=-1).cpu().numpy()
|
| 115 |
scores = list(probs.flatten())
|
| 116 |
|
| 117 |
-
sorted_sl = sorted(zip(scores,
|
| 118 |
|
| 119 |
pred["labels"] = textlet
|
| 120 |
pred["scores"], pred["labels"] = zip(*sorted_sl)
|
|
|
|
| 114 |
probs = probs.softmax(dim=-1).cpu().numpy()
|
| 115 |
scores = list(probs.flatten())
|
| 116 |
|
| 117 |
+
sorted_sl = sorted(zip(scores, labels), key=lambda t:t[0], reverse=True)
|
| 118 |
|
| 119 |
pred["labels"] = textlet
|
| 120 |
pred["scores"], pred["labels"] = zip(*sorted_sl)
|