Update tasks/text.py

tasks/text.py

@@ -67,8 +67,12 @@ async def evaluate_text(request: TextEvaluationRequest):
             device="cpu"
             )
     
-    predictions = [label2id[classifier(quote)[0]["label"]] for quote in test_dataset["quote"]]
-    print(predictions)
+    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)
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE