Update tasks/text.py

tasks/text.py

@@ -59,13 +59,21 @@ async def evaluate_text(request: TextEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     
 
- 
-    # load model from huggingface.co/models using our repository id
+    # Load the model pipeline
     classifier = pipeline(
         task="text-classification", 
         model="dvilasuero/ModernBERT-frugal", 
-        device=0,
+        device=0,  # Ensure this points to the correct device (GPU or CPU)
     )
+    
+    # Extract texts from the test dataset for inference
+    texts = [example["text"] for example in test_dataset]
+    
+    # Perform predictions using the model
+    raw_predictions = classifier(texts)
+    
+    # Convert raw predictions to numeric labels
+    predictions = [LABEL_MAPPING[pred["label"]] for pred in raw_predictions]
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE