Update README.md

README.md

@@ -59,7 +59,7 @@ from melspectrogram import MelSpectrogramProcessor  # Import Mel spectrogram pro
 # Initialize the preprocessor and model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 preprocessor = MelSpectrogramProcessor(device=device)
-model = cvt13().to(device)
+model = cvt13()
 
 # Load weights trained using Cross-Entropy
 model.load_state_dict(torch.load("ce.pth", map_location="cpu")['encoder'])
@@ -74,6 +74,7 @@ model.load_state_dict(torch.load("supcon.pth", map_location="cpu"))
 model.load_state_dict(torch.load("protoclr.pth", map_location="cpu"))
 
 # Optional: Move the model to GPU for faster processing if available using : model = model.to('cuda') , for instance.
+model = model.to(device)
 model.eval()
 
 # Load and preprocess a sample audio waveform