Spaces:
Runtime error
Runtime error
Commit
·
570c67b
1
Parent(s):
5adbb2e
update
Browse files- app/main.py +1 -1
app/main.py
CHANGED
@@ -108,7 +108,7 @@ def classify_multiple(data, model, tokenizer, device):
|
|
108 |
outputs = model(test_ids.to(device), test_attention_mask.to(device))
|
109 |
result = torch.argmax(outputs, dim= -1)
|
110 |
result_label = get_key(result.item(), LABELS)
|
111 |
-
|
112 |
|
113 |
return result_list
|
114 |
|
|
|
108 |
outputs = model(test_ids.to(device), test_attention_mask.to(device))
|
109 |
result = torch.argmax(outputs, dim= -1)
|
110 |
result_label = get_key(result.item(), LABELS)
|
111 |
+
result_list.append(result_label)
|
112 |
|
113 |
return result_list
|
114 |
|