update demo

demo.py

@@ -85,14 +85,13 @@ def generate_new_experiment() -> Experiment:
     imgs_to_sample = (NUMBER_OF_IMAGES_PER_ROW * NUMBER_OF_ROWS) // 2
 
     corrupted_image = {"name": str(random.choice(list(image_id.glob("*corrupted*"))))}
-    corrupted_ending_index = corrupted_image["name"].split(".")[0].split("_")[-1]
     sdedit_images = [
-        {"name": str(img), "algo": f"SDEdit"}
-        for img in random.sample(list((image_id / "sde").glob(f"*_{corrupted_ending_index}*")), imgs_to_sample)
+        {"name": str(img), "algo": "SDEdit"}
+        for img in random.sample(list((image_id / "sde").glob(f"*")), imgs_to_sample)
     ]
     odedit_images = [
-        {"name": str(img), "algo": f"ODEdit"}
-        for img in random.sample(list((image_id / "ode").glob(f"*_{corrupted_ending_index}*")), imgs_to_sample)
+        {"name": str(img), "algo": "ODEdit"}
+        for img in random.sample(list((image_id / "ode").glob(f"*")), imgs_to_sample)
     ]
     total_images = sdedit_images + odedit_images
     random.shuffle(total_images)