Update tasks/text.py

tasks/text.py

@@ -58,7 +58,16 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     # Make random predictions (placeholder for actual model inference)
     true_labels = test_dataset["label"]
-    predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
+    config = AutoConfig.from_pretrained("camillebrl/ModernBERT-envclaims-overfit")
+        label2id = config.label2id
+        classifier = pipeline(
+            "text-classification",
+            "camillebrl/ModernBERT-envclaims-overfit",
+            device="cpu"
+            )
+    
+    predictions = [label2id[classifier(quote)[0]["label"]] for quote in test_dataset["quote"]]
+    print(predictions)
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE