Update demo.py

demo.py

@@ -81,7 +81,7 @@ def experiment_to_dict(experiment, skip=False):
     
 def generate_new_experiment() -> Experiment:
     wanted_corruptions = ["spatter", "impulse_noise", "speckle_noise", "gaussian_noise", "pixelate", "jpeg_compression"]
-    corruption = random.choice([f for f in list(Path(f"./images/{DATASET}").glob("*/*")) if f.is_dir() and f.name in wanted_corruptions)
+    corruption = random.choice([f for f in list(Path(f"./images/{DATASET}").glob("*/*")) if f.is_dir() and f.name in wanted_corruptions])
     image_id = random.choice(list(corruption.glob("*")))
     imgs_to_sample = (NUMBER_OF_IMAGES_PER_ROW * NUMBER_OF_ROWS) // 2