Spaces:
Sleeping
Sleeping
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)
|