Update README.md

README.md

@@ -19,7 +19,7 @@ This model is a self-play fine-tuned diffusion model at iteration 3 from [runway
 
 ### Model Description
 
-- Model type: An diffusion model with unet fine-tuned, based on the strucure of stable diffusion 1.5
+- Model type: A diffusion model with unet fine-tuned, based on the structure of stable diffusion 1.5
 - Language(s) (NLP): Primarily English
 - License: Apache-2.0
 - Finetuned from model: runwayml/stable-diffusion-v1-5
@@ -38,7 +38,29 @@ The following hyperparameters were used during training:
 - lr_warmup_steps: 200
 - num_training_steps: 500
 
-  
+### Usage
+To use the model, you must first load the SD1.5 base model and then substitute its unet with our fine-tuned version.
+```python
+from diffusers import StableDiffusionPipeline, UNet2DConditionModel
+import torch
+
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+
+unet_id = "UCLA-AGI/SPIN-Diffusion-iter3"
+unet = UNet2DConditionModel.from_pretrained(unet_id, subfolder="unet", torch_dtype=torch.float16)
+pipe.unet = unet
+
+###The rest of your generation code
+```
+
+### Evaluation Results on [Pick-a-pic test set](https://huggingface.co/datasets/yuvalkirstain/pickapic_v2/viewer/default/test_unique)
+| Metric                |     Best of Five                 | Mean | Median
+|-----------------------|---------------------------|--------|-------
+| HPS         | 0.28          | 0.27 | 0.27
+| Aesthetic   | 6.26    | 5.94 | 5.98
+| Image Reward         | 1.13    | 0.53 | 0.67
+| Pickapic Score   | 22.00   |21.36 | 21.46
 ## Citation
 ```
 @misc{yuan2024self,