Update README.md

README.md

@@ -13,7 +13,15 @@ It is the fusion of Stable Diffusion 1.4 and a LORA fine-tuned on 28218 patholog
 
 ### Model Description
 
-<!-- Provide a longer summary of what this model is. -->
+The model is trained on Stable Diffusion 1.4 with a UNet architecture and a pretrained VAE. It uses a batch size of 16 and lora rank of 32.
 
-- **Developed by:** [Muyu He]
-- **Finetuned from model [optional]:** [CompVis/stable-diffusion-v1-4]
+### How to use
+
+```python
+pipe = DiffusionPipeline.from_pretrained(
+  "RiddleHe/SD14_pathology_base", torch_dtype=torch.float16
+)
+pipe.to('cuda')
+
+prompt = "A histopathology image of breast cancer tissue"
+```