Update tasks/audio.py

tasks/audio.py

@@ -61,7 +61,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     model, device = load_model(model_path)
 
     def preprocess_audio(example, target_length=32000):
-            """
+        """
         Convert dataset into tensors:
         - Convert to tensor
         - Normalize waveform
@@ -94,13 +94,13 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     predictions = []
 
     with torch.no_grad():
-    for waveforms, labels in train_loader:
-        waveforms, labels = waveforms.to(device), labels.to(device)
-        
-        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())
+        for waveforms, labels in train_loader:
+            waveforms, labels = waveforms.to(device), labels.to(device)
+            
+            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())
     
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE