Spaces:
Build error
Build error
update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def get_sentiments(text):
|
|
| 52 |
logits = model(**encoded_input).logits
|
| 53 |
|
| 54 |
logits_list = logits[0].tolist()
|
| 55 |
-
logits_labels = [model.config.id2label[i]
|
| 56 |
for i in range(0, len(logits_list))]
|
| 57 |
|
| 58 |
probs = softmax(logits_list)
|
|
|
|
| 52 |
logits = model(**encoded_input).logits
|
| 53 |
|
| 54 |
logits_list = logits[0].tolist()
|
| 55 |
+
logits_labels = [model.config.id2label[i + 1]
|
| 56 |
for i in range(0, len(logits_list))]
|
| 57 |
|
| 58 |
probs = softmax(logits_list)
|