Update README.md

README.md

@@ -44,6 +44,7 @@ device = torch.device("cuda")
 torch_dtype = torch.float16 if device.type in ["mps", "cuda"] else torch.float32
 pipeline = StableDiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype).to(device)
 image = pipeline("PROMPT GOES HERE").images[0]
+image.save("output.png")
 ```
 
 #### Limitations and bias