update

tasks/text.py

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