Update tasks/audio.py

tasks/audio.py

@@ -209,9 +209,9 @@ async def evaluate_audio(request: AudioEvaluationRequest):
         for data, target in test_loader:
             output = model(data).squeeze()
             pred = torch.argmax(output, dim=-1)
-            print('pred:', pred)
             predictions.extend(pred.tolist())
 
+    print("predictions:", predictions)
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE
     #--------------------------------------------------------------------------------------------