Update README.md

README.md

@@ -1,4 +1,3 @@
-
 ---
 license: creativeml-openrail-m
 base_model: CompVis/stable-diffusion-v1-4
@@ -9,13 +8,16 @@ tags:
 - diffusers
 - lora
 inference: true
+datasets:
+- mozci/tinysketch
+library_name: diffusers
+pipeline_tag: text-to-image
 ---
     
 # LoRA text2image fine-tuning - mozci/nonstablediff
-These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. The weights were fine-tuned on the mozci/tinysketch dataset. You can find some example images in the following. 
+Stable diffusion fine-tuning to achieve simple sketches as the outputs. These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. The weights were fine-tuned on the mozci/tinysketch dataset. You can find some example images in the following. 
 
 ![img_0](./image_0.png)
 ![img_1](./image_1.png)
 ![img_2](./image_2.png)
-![img_3](./image_3.png)
-
+![img_3](./image_3.png)