Update tasks/audio.py

tasks/audio.py

@@ -108,7 +108,10 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     predictions = []
     with torch.inference_mode():
+        model.eval()
+        model.half()
         for data, target in test_loader:
+            data = data.half()
             output = model(data).squeeze()
             pred = torch.argmax(output, dim=-1)
             predictions.extend(pred.tolist())