Update tasks/audio.py

tasks/audio.py

@@ -4,8 +4,8 @@ from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import random
 import os
-import torch
-from torch.utils.data import DataLoader
+#import torch
+#from torch.utils.data import DataLoader
 #from /app/tasks/Model_Loader.py import M5, load_model
 
 from .utils.evaluation import AudioEvaluationRequest
@@ -57,8 +57,8 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     #--------------------------------------------------------------------------------------------   
     
     # Make random predictions (placeholder for actual model inference)
-    model_path = "quantized_teacher_m5_static.pth"
-    model, device = load_model(model_path)
+    #model_path = "quantized_teacher_m5_static.pth"
+    #model, device = load_model(model_path)
 
     # def preprocess_audio(example, target_length=32000):
     #     """