update

tasks/text.py

@@ -98,13 +98,9 @@ async def evaluate_text(request: TextEvaluationRequest):
             tokenized_inputs = tokenizer(batch_quotes, padding=True, truncation=True, return_tensors='pt').to(device)
             
             # Forward pass through the model
-            outputs = model(**tokenized_inputs)
-    with torch.no_grad():
-        print('BEFORE PREDICTION')
-        
-        outputs = model(**tokenized_test)
-        p = torch.argmax(outputs.logits, dim=1)
-        predictions = np.append(predictions, p.cpu().numpy())
+            p = model(**tokenized_inputs)
+
+            predictions = np.append(predictions, p.cpu().numpy())
 
     print("Finished prediction run")