Update demo.py

demo.py

@@ -55,7 +55,7 @@ def load_surgvlp_encoder(device):
     """
     config_path = './utils/config_surgvlp.py'
     configs = Config.fromfile(config_path)['config']
-    encoder_model, encoder_preprocess = surgvlp.load(configs.model_config, device=device, pretrain='SurgVLP.pth')
+    encoder_model, encoder_preprocess = surgvlp.load(configs.model_config, device=device, pretrain='SurgVLP2.pth')
     encoder_model.eval()
     return encoder_model, encoder_preprocess
 
@@ -80,8 +80,8 @@ def main():
     
     # Define the directories containing images and corresponding label files.
     global image_files
-    images_dir = "./test_data/images/VID22/"
-    labels_dir = "./test_data/labels/VID22/"
+    images_dir = "./test_data/images/VID01/"
+    labels_dir = "./test_data/labels/VID01/"
     image_files = [os.path.join(images_dir, f) for f in sorted(os.listdir(images_dir)) if f.lower().endswith('.png')]
     random.shuffle(image_files)
     # Get first 20 images.