update

tasks/text.py

@@ -90,8 +90,8 @@ async def evaluate_text(request: TextEvaluationRequest):
     with torch.no_grad():
         print('BEFORE PREDICTION')
         
-        test_input_ids = tokenized_test["input_ids"].to(device)
-        test_attention_mask = tokenized_test["attention_mask"].to(device)
+        test_input_ids = tokenized_test["input_ids"]
+        test_attention_mask = tokenized_test["attention_mask"]
         outputs = model(test_input_ids, test_attention_mask)
         p = torch.argmax(outputs.logits, dim=1)
         predictions = np.append(predictions, p.cpu().numpy())