Update tasks/audio.py

tasks/audio.py

@@ -84,6 +84,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     # Split dataset
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
+    test_dataset = test_dataset[:10]
     
     # Start tracking emissions
     tracker.start()
@@ -101,8 +102,9 @@ async def evaluate_audio(request: AudioEvaluationRequest):
         out = model(batch[0])
         out_logits = torch.sigmoid(out)
         predictions.extend(out_logits.round().tolist())
-
+        
     # Make random predictions (placeholder for actual model inference)
+    predictions = [x[0] for x in predictions]
     true_labels = test_dataset["label"]