Create README.md

README.md

@@ -0,0 +1,27 @@
+# Unconditional Pothole Images Generation 
+
+## Generating New Images
+
+```{python}
+from diffusers import DDPMPipeline
+import torch
+from PIL import Image
+
+pipe = DDPMPipeline.from_pretrained("ridzy619/pothole-unconditional-generation")
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe.to(device)
+
+images = pipe(batch_size=16).images
+
+def make_grid(images, rows, cols):
+    w, h = images[0].size
+    grid = Image.new('RGB', size=(cols*w, rows*h))
+    for i, image in enumerate(images):
+        grid.paste(image, box=(i%cols*w, i//cols*h))
+    return grid
+
+images_grid = make_grid(images, 4, 4)
+images_grid
+
+```