update

tasks/text.py

@@ -99,8 +99,9 @@ async def evaluate_text(request: TextEvaluationRequest):
             
             # Forward pass through the model
             p = model(**tokenized_inputs)
+            output = torch.argmax(p.logits, dim=1).cpu().numpy()
             print(p)
-            predictions = np.append(predictions, p.numpy())
+            predictions = np.append(predictions, output)
 
     print("Finished prediction run")