Update README.md

README.md

@@ -39,18 +39,18 @@ from datasets import Dataset
 from pathlib import Path
 from PIL import Image
 
-face_synthetics = Path("images/original")
-canny_synthetics = Path("images/canny_images")
+face_image = Path("images/original")
+canny_image = Path("images/canny_images")
 
 def entry_for_id(entry_id):
     if type(entry_id) == int:
         entry_id = f"{entry_id:05}"
-    image = Image.open(face_synthetics/f"fractal_{entry_id}.jpg")
-    image_seg = Image.open(canny_synthetics/f"canny_fractal_{entry_id}.jpg")
+    image = Image.open(face_image/f"fractal_{entry_id}.jpg")
+    image_cond = Image.open(canny_image/f"canny_fractal_{entry_id}.jpg")
     caption = "gothic fractals"
     return {
         "image": image,
-        "canny_images": image_seg,
+        "canny_images": image_cond,
         "caption": caption,
     }