Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import numpy as np
 import librosa
 import time
-from src.utility import slice_songs  # Adjust your imports as needed
+from src.utility import slice_songs
 from src.models import MusicCNN, MusicCRNN2D
 
 # Configuration
@@ -12,9 +12,9 @@ HOP_LENGTH = 512
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 NUM_CLASSES = 10
 
-# Model loading (example for cnn and crnn; update paths as necessary)
+# Model loading
 models = {"cnn": {}, "crnn": {}}
-# For instance:
+
 cnn_model_paths = {1: "models/cnn/1s.pth", 3: "models/cnn/3s.pth", 5: "models/cnn/5s.pth", 10: "models/cnn/10s.pth"}
 crnn_model_paths = {1: "models/crnn/1s.pth", 3: "models/crnn/3s.pth", 5: "models/crnn/5s.pth",
                     10: "models/crnn/10s.pth"}