Update tasks/text.py

tasks/text.py

@@ -4,6 +4,8 @@ from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import random
 
+from transformers import pipeline
+
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
@@ -56,9 +58,14 @@ async def evaluate_text(request: TextEvaluationRequest):
     # Update the code below to replace the random baseline by your model inference within the inference pass where the energy consumption and emissions are tracked.
     #--------------------------------------------------------------------------------------------   
     
-    # Make random predictions (placeholder for actual model inference)
-    true_labels = test_dataset["label"]
-    predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
+
+ 
+    # load model from huggingface.co/models using our repository id
+    classifier = pipeline(
+        task="text-classification", 
+        model="dvilasuero/ModernBERT-frugal", 
+        device=0,
+    )
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE