Clean up

tasks/text.py

@@ -11,7 +11,17 @@ router = APIRouter()
 
 DESCRIPTION = "Most common class baseline"
 ROUTE = "/text"
-    
+
+def baseline_model(dataset_length):
+    # Make random predictions (placeholder for actual model inference)
+    #predictions = [random.randint(0, 7) for _ in range(dataset_length)]
+
+    # My favorate baseline is the most common class.
+    predictions = [0] * dataset_length
+
+return predictions
+
+
 @router.post(ROUTE, tags=["Text Task"], 
              description=DESCRIPTION)
 async def evaluate_text(request: TextEvaluationRequest):
@@ -55,15 +65,10 @@ async def evaluate_text(request: TextEvaluationRequest):
     # YOUR MODEL INFERENCE CODE HERE
     # 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))]
 
-    # My favorate baseline is the most common class.
     true_labels = test_dataset["label"]
-    predictions = [0] * len(true_labels)
-
+    predictions = baseline_model(len(true_labels))
+    
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE
     #--------------------------------------------------------------------------------------------