Update README.md

README.md

@@ -18,7 +18,7 @@ This is a Stable Diffusion model fine-tuned on the sbtstyle concept with DreamBo
 
 This model was created as part of the DreamBooth Hackathon 🔥. Visit the [organisation page](https://huggingface.co/dreambooth-hackathon) for instructions on how to take part!
 
-## Examples
+## Examples and data
 
 <table>
   <tr>
@@ -40,6 +40,23 @@ This model was created as part of the DreamBooth Hackathon 🔥. Visit the [orga
   </tr>
 </table>
 
+## Dataset
+
+The full dataset used to train on can be found here: [`Sanderbaduk/sobotta-anatomical-dataset`](https://huggingface.co/datasets/Sanderbaduk/sobotta-anatomical-dataset).
+These images and others like it can be found on wikimedia commons.
+
+If you use this directly you will get an error as the raw images have different numbers of channels. Include a grayscale step to fix this and ensure that classic feel.
+```
+self.transforms = transforms.Compose(
+    [
+        transforms.Resize(size),
+        transforms.Grayscale(num_output_channels=3), 
+        transforms.CenterCrop(size),
+        transforms.ToTensor(),
+        transforms.Normalize([0.5], [0.5]),
+    ]
+)
+```
 
 ## Usage