Upload README.md with huggingface_hub

README.md

@@ -1,20 +1,45 @@
 ---
-library_name: keras
+license: creativeml-openrail-m
+tags:
+- keras
+- keras-cv
+- diffusers
+- stable-diffusion
+- text-to-image
+- diffusion-models-class
+- dreambooth
+- nature
+widget:
+- text: a photo of puggieace dog on the beach
 ---
 
-## Model description
+# DreamBooth model for the `puggieace` concept trained by nielsgl on the nielsgl/dreambooth-ace dataset.
 
-More information needed
+This is a KerasCV Stable Diffusion V2.1 model fine-tuned on the puggieace concept with DreamBooth. It can be used by modifying the `instance_prompt`: **a photo of puggieace**
 
-## Intended uses & limitations
+This model was created as part of the Keras DreamBooth Sprint 🔥. Visit the [organisation page](https://huggingface.co/keras-dreambooth) for instructions on how to take part!
 
-More information needed
+## Description
 
-## Training and evaluation data
 
-More information needed
+This is a KerasCV Stable Diffusion model fine-tuned on `dog` images for the nature theme.
 
-## Training procedure
+
+## Usage
+
+```python
+from huggingface_hub import from_pretrained_keras
+import keras_cv
+import matplotlib.pyplot as plt
+
+
+model = keras_cv.models.StableDiffusionV2(img_width=512, img_height=512, jit_compile=True)
+model._diffusion_model = from_pretrained_keras(nielsgl/dreambooth-pug-ace-sd2.1-base)
+model._text_encoder = from_pretrained_keras(nielsgl/dreambooth-pug-ace-sd2.1-base-text-encoder)
+
+images = model.text_to_image("a photo of puggieace dog on the beach", batch_size=3)
+plt.imshow(image[0])
+```
 
 ### Training hyperparameters
 
@@ -38,4 +63,3 @@ The following hyperparameters were used during training:
 | epsilon | 1e-07 |
 | centered | False |
 | training_precision | float32 |
-