updated demo

.gitignore

@@ -0,0 +1 @@
+unfiltered_images

demo.py

@@ -80,7 +80,7 @@ def experiment_to_dict(experiment, skip=False):
     return info
     
 def generate_new_experiment() -> Experiment:
-    corruption = random.choice(list(Path(f"./images/{DATASET}").glob("*")))
+    corruption = random.choice([f for f in list(Path(f"./images/{DATASET}").glob("*/*")) if f.is_dir()])
     image_id = random.choice(list(corruption.glob("*")))
     imgs_to_sample = (NUMBER_OF_IMAGES_PER_ROW * NUMBER_OF_ROWS) // 2
 
@@ -95,6 +95,7 @@ def generate_new_experiment() -> Experiment:
         for img in random.sample(list((image_id / "ode").glob(f"*_{corrupted_ending_index}*")), imgs_to_sample)
     ]
     total_images = sdedit_images + odedit_images
+    random.shuffle(total_images)
 
     return Experiment(
         DATASET,