fixes

tasks/text.py

@@ -11,7 +11,7 @@ router = APIRouter()
 
 DESCRIPTION = "Most common class baseline"
 ROUTE = "/text"
-
+    
 @router.post(ROUTE, tags=["Text Task"], 
              description=DESCRIPTION)
 async def evaluate_text(request: TextEvaluationRequest):
@@ -61,6 +61,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     #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)
 
     #--------------------------------------------------------------------------------------------