Update README.md

README.md

@@ -22,6 +22,20 @@ The file `galaxsd_final.safetensors` contains the fine-tuned Stable Diffusion we
 pip install diffusers transformers accelerate safetensors
 ```
 
+2. Load and run the model:
+```bash
+from diffusers import StableDiffusionPipeline
+import torch
+
+pipe = StableDiffusionPipeline.from_single_file(
+    "galaxsd_final.safetensors",
+    torch_dtype=torch.float16
+).to("cuda")
+
+prompt = "sdss, spiral galaxy, loosely wound spiral arms, 2 spiral arms"
+image = pipe(prompt).images[0]
+image.save("output.png")
+```
 
 
 ## 🔗 Project Resources