Update utils.py

utils.py

@@ -16,7 +16,7 @@ def get_text(title: str, abstract: str):
 def get_labels(text, model, tokenizer, count_labels=8):
   tokens = tokenizer(text, return_tensors='pt')
   outputs = model(**tokens)
-  probs = torch.nn.Softmax(dim=count_labels)(outputs.logits)
+  probs = torch.nn.Softmax(dim=1)(outputs.logits)
   
   labels = ['Computer_science', 'Economics',
        'Electrical_Engineering_and_Systems_Science', 'Mathematics',