24000 model

script.py

@@ -61,8 +61,8 @@ model = LCNN(return_emb=False).to(device)
 # model_path = './checkpoints/LCNN_ALL_DATA_HI_FREQ_22050.pth'
 
 # model_path = './checkpoints/LCNN_ALL_DATA_FS_16000.pth'
-model_path = './checkpoints/LCNN_ALL_DATA_FS_22050.pth'
-# model_path = './checkpoints/LCNN_ALL_DATA_FS_24000.pth'
+# model_path = './checkpoints/LCNN_ALL_DATA_FS_22050.pth'
+model_path = './checkpoints/LCNN_ALL_DATA_FS_24000.pth'
 
 model.load_state_dict(torch.load(model_path, map_location=device))
 
@@ -133,7 +133,7 @@ for el in tqdm.tqdm(dataset_remote):
 
     # RUNNING ON HUGGINGFACE
     file_like = io.BytesIO(el["audio"]["bytes"])
-    tensor, sr = preprocess(file_like, target_sr=22050)
+    tensor, sr = preprocess(file_like, target_sr=24000)
     # # RUNNING LOCALLY
     # tensor = preprocess(el)