Update README.md

README.md

@@ -50,13 +50,13 @@ tags:
 should probably proofread and complete it, then remove this comment. -->
 
 
-# SDXL LoRA DreamBooth - Prgckwb/trpfrog-sdxl-lora-3
+# SDXL LoRA DreamBooth - Prgckwb/trpfrog-sdxl-lora
 
 <Gallery />
 
 ## Model description
 
-These are Prgckwb/trpfrog-sdxl-lora-3 LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
+These are Prgckwb/trpfrog-sdxl-lora LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
 
 The weights were trained  using [DreamBooth](https://dreambooth.github.io/).
 
@@ -77,10 +77,14 @@ import torch
 base_model_id = 'stabilityai/stable-diffusion-xl-base-1.0'
 lora_model_id = 'Prgckwb/trpfrog-sdxl-lora'
 
-pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = DiffusionPipeline.from_pretrained(
+  base_model_id, torch_dtype=torch.float16
+).to("cuda")
 pipe.load_lora_weights(lora_model_id)
 
-image = pipe("trpfrog", num_inference_steps=25).images[0]
+image = pipe(
+  "an icon of trpfrog",
+  num_inference_steps=25
+).images[0]
 image.save('trpfrog.png')
 ```