Update README.md

README.md

@@ -18,7 +18,27 @@ should probably proofread and complete it, then remove this comment. -->
 
 # LoRA text2image fine-tuning - BogdanTurbal/synt_data_v_4
 
-These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were fine-tuned on the BogdanTurbal/gen_data_test dataset. You can find some example images in the following. 
+
+
+These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. 
+Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images.
+
+With this model you can generate MRI of the brain of 4 classes:
+1) NonDemented
+2) VeryMildDemented
+3) MildDemented
+4) ModerateDemeneted
+
+For generation use following corresponding prompts (or any other modification)
+
+1) MRI of NonDemented brain
+2) MRI of VeryMildDemented brain
+3) MRI of MildDemented brain
+4) MRI of ModerateDemeneted brain
+
+
+
+You can find some example images in the following. 
 
 ![img_0](./image_0.png)
 ![img_1](./image_1.png)