ready

tasks/text.py

@@ -79,6 +79,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     # Split dataset
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
+    test_df = pd.DataFrame(test_dataset)
     
     # Start tracking emissions
     tracker.start()
@@ -89,7 +90,6 @@ 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.
     #--------------------------------------------------------------------------------------------   
 
-    test_df = pd.DataFrame(dataset['test'])
     
     #download model
     download(repo_id = "kantundpeterpan/frugal-ai-toy", dst = "skops_test")