Update README.md

README.md

@@ -13,6 +13,20 @@ tags:
 
 ![So cute, right?](https://huggingface.co/nroggendorff/cats/resolve/main/samples/0024.png)
 
+### Diffusers
+```py
+from diffusers import DiffusionPipeline
+
+pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
+pipe = pipe.to("cuda")
+
+image = pipe().images[0]  
+
+image.save("cat.png")
+```
+
+
+### Model Details
 - `train_batch_size`: 16
 - `eval_batch_size`: 16
 - `num_epochs`: 50
@@ -20,4 +34,19 @@ tags:
 - `learning_rate`: 1e-4
 - `lr_warmup_steps`: 500
 - `mixed_precision`: "fp16"
-- `eval_metric`: "mean_squared_error"
+- `eval_metric`: "mean_squared_error"
+
+### Bias
+- This model may exhibit biases due to its training data. It will not display images of abused or sick cats, as it prioritizes the well-being of animals.
+
+### Limitations
+
+- The model does not achieve perfect photorealism
+- The model cannot render legible text
+- The model was trained on a medium-to-large-scale dataset
+  [Cats vs Dogs](https://www.kaggle.com/datasets/shaunthesheep/microsoft-catsvsdogs-dataset/data)
+  
+### Developed by
+- Noa Linden Roggendorff
+
+*This model card was written by Noa Roggendorff and is based on the [Stable Diffusion v1-5 Model Card](https://huggingface.co/runwayml/stable-diffusion-v1-5).*