Update demo.py

demo.py

@@ -85,8 +85,8 @@ def main():
     
     # Define the directories containing images and corresponding label files.
     global image_files
-    images_dir = "./test_data/images/VID01"
-    labels_dir = "./test_data/labels/VID01/"
+    images_dir = "./test_data/images/VID"
+    labels_dir = "./test_data/labels/VID/"
     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)