Update README.md

README.md

@@ -17,8 +17,6 @@ widget:
 
 This is a Stable Diffusion model fine-tuned on the trpfrog concept with DreamBooth. It can be used by modifying the `instance_prompt`: **a photo of trpfrog**
 
-This model was created as part of the DreamBooth Hackathon 🔥. Visit the [organisation page](https://huggingface.co/dreambooth-hackathon) for instructions on how to take part!
-
 ## Description
 TrpFrog ([@Trpfrog](https://twitter.com/trpfrog?s=21&t=GcDGy74adYhOBYxX9HloOg)) is a famous Japanese Twitter comedian.
 
@@ -44,7 +42,8 @@ pipe = StableDiffusionPipeline.from_pretrained(
     model_id, torch_dtype=torch.float16
 ).to(device)
 
-image = pipeline().images[0]
+image = pipe(prompt).images[0]
+image.save("trpfrog.jpg")
 ```
 
 ## Generated Images