Update tasks/audio.py

tasks/audio.py

@@ -72,6 +72,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
 
     
     true_labels = test_dataset["label"]
+    predictions = []
 
     with torch.no_grad():
     for waveforms, labels in test_loader:
@@ -80,10 +81,8 @@ async def evaluate_audio(request: AudioEvaluationRequest):
         # Run Model
         outputs = model(waveforms)
         predicted_label = torch.argmax(F.softmax(outputs, dim=1), dim=1)
-
         true_labels.extend(labels.cpu().numpy())
         predicted_labels.extend(predicted_label.cpu().numpy())
-    predictions = [random.randint(0, 1) for _ in range(len(true_labels))]
     
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE