Update tasks/audio.py

tasks/audio.py

@@ -6,7 +6,7 @@ import os
 import torch
 from torch.utils.data import DataLoader, TensorDataset
 from torchaudio import transforms
-from torchvision import models
+#from torchvision import models
 import onnxruntime as ort  # Add ONNX Runtime
 from .utils.evaluation import AudioEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info