Update README.md

README.md

@@ -1,14 +1,34 @@
 ---
 license: mit
+language: en
+tags:
+  - text-to-image
 ---
-### zach fox on Stable Diffusion via Dreambooth
-#### model by nateraw
-This your the Stable Diffusion model fine-tuned the zach fox concept taught to Stable Diffusion with Dreambooth.
+
+# Zach Fox Stable Diffusion
+
+Stable Diffusion v1.5 model fine-tuned to generate pictures of Zach Fox with Dreambooth.
+
 It can be used by modifying the `instance_prompt`: **a photo of sks zach fox**
 
 You can also train your own concepts and upload them to the library by using [this notebook](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/sd_dreambooth_training.ipynb).
 And you can run your new concept via `diffusers`: [Colab Notebook for Inference](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/sd_dreambooth_inference.ipynb), [Spaces with the Public Concepts loaded](https://huggingface.co/spaces/sd-dreambooth-library/stable-diffusion-dreambooth-concepts)
 
+
+## Usage
+
+To use this model, you can run the following...
+
+```python
+import torch
+from diffusers import StableDiffusionPipeline
+
+pipe = StableDiffusionPipeline.from_pretrained("nateraw/zach-fox", torch_dtype=torch.float16).to("cuda")
+
+prompt = "a photo of sks zach fox, oil on canvas"
+image = pipe(prompt).images[0]
+```
+
 Here are the images used for training this concept:
 ![image 0](https://huggingface.co/nateraw/zach-fox-v2/resolve/main/concept_images/8.jpeg)
 ![image 1](https://huggingface.co/nateraw/zach-fox-v2/resolve/main/concept_images/13.jpeg)