Update tasks/text.py

tasks/text.py

@@ -66,14 +66,14 @@ async def evaluate_text(request: TextEvaluationRequest):
             "camillebrl/ModernBERT-envclaims-overfit",
             device="cpu"
             )
-    
+    print("len dataset : ", len(test_dataset["quote"]))
     predictions = []
     for batch in range(0, len(test_dataset["quote"]), 32):  # Ajustez la taille des batchs
         batch_quotes = test_dataset["quote"][batch:batch + 32]
         batch_predictions = classifier(batch_quotes)
         predictions.extend([label2id[pred["label"]] for pred in batch_predictions])
         print(predictions)
-
+    print("final predictions : ", predictions)
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE
     #--------------------------------------------------------------------------------------------   
@@ -84,6 +84,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     # Calculate accuracy
     accuracy = accuracy_score(true_labels, predictions)
+    print("accuracy : ", accuracy)
     
     # Prepare results dictionary
     results = {
@@ -102,5 +103,7 @@ async def evaluate_text(request: TextEvaluationRequest):
             "test_seed": request.test_seed
         }
     }
+
+    print("results : ", results)
     
     return results