Update README.md

README.md

@@ -35,7 +35,32 @@ prompt: gothic fractals
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
+from datasets import Dataset
+from pathlib import Path
+from PIL import Image
+
+face_synthetics = Path("images/original")
+canny_synthetics = 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")
+    caption = "gothic fractals"
+    return {
+        "image": image,
+        "canny_images": image_seg,
+        "caption": caption,
+    }
+
+
+def generate_entries():
+    for x in range(1, 6463):  # Start from 1 and go up to 6462
+        yield entry_for_id(x)
+
+ds = Dataset.from_generator(generate_entries)
+ds.push_to_hub('Jieya/fractal_image_6462')
 ```
 
 #### Limitations and bias