kadabengaran commited on
Commit
570c67b
·
1 Parent(s): 5adbb2e
Files changed (1) hide show
  1. 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
- result.append(result_label)
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