Update README.md

README.md

@@ -21,8 +21,9 @@ This model was created as part of the DreamBooth Hackathon 🔥. Visit the [orga
 
 ## Description
 
+This is a fine-tuned model of Stable Diffusion with the powerful help of Dreambooth. 
+The model is made specifically for TrpFrog ([@Trpfrog](https://twitter.com/trpfrog?s=21&t=GcDGy74adYhOBYxX9HloOg)) and faithfully reproduces its appearance.
 
-This is a Stable Diffusion model fine-tuned on `` images for the wildcard theme.
 
 
 ## Usage
@@ -30,7 +31,21 @@ This is a Stable Diffusion model fine-tuned on `` images for the wildcard theme.
 ```python
 from diffusers import StableDiffusionPipeline
 
-pipeline = StableDiffusionPipeline.from_pretrained('Prgckwb/trpfrog-diffusion')
+keyword = "trpfrog"
+prompt = f"a photo of {keyword}"
+
+model_id = "Prgckwb/trpfrog-diffusion"
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id, torch_dtype=torch.float16
+).to(device)
+
 image = pipeline().images[0]
-image
-```
+```
+
+## Generated Images
+**a photo of trpfrog eating kiwi**
+![](output.png)
+
+**a photo of trpfrog growing shiitake mushrooms**
+![](output2.png)