Update README.md

README.md

@@ -42,13 +42,13 @@ tags:
 should probably proofread and complete it, then remove this comment. -->
 
 
-# SD3 DreamBooth LoRA - Prgckwb/trpfrog-sd3.5m-lora
+# SD3 DreamBooth LoRA - Prgckwb/trpfrog-sd3.5-medium-lora
 
 <Gallery />
 
 ## Model description
 
-These are Prgckwb/trpfrog-sd3.5m-lora DreamBooth LoRA weights for stabilityai/stable-diffusion-3.5-medium.
+These are Prgckwb/trpfrog-sd3.5-medium-lora DreamBooth LoRA weights for stabilityai/stable-diffusion-3.5-medium.
 
 The weights were trained using [DreamBooth](https://dreambooth.github.io/) with the [SD3 diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/README_sd3.md).
 
@@ -60,7 +60,7 @@ You should use `an icon of trpfrog` to trigger the image generation.
 
 ## Download model
 
-[Download the *.safetensors LoRA](Prgckwb/trpfrog-sd3.5m-lora-2/tree/main) in the Files & versions tab.
+[Download the *.safetensors LoRA](Prgckwb/trpfrog-sd3.5-medium-lora/tree/main) in the Files & versions tab.
 
 ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
 
@@ -69,19 +69,21 @@ from diffusers import AutoPipelineForText2Image
 import torch
 
 pipeline = AutoPipelineForText2Image.from_pretrained(
-    stabilityai/stable-diffusion-3.5-medium,
+    'stabilityai/stable-diffusion-3.5-medium', 
     torch_dtype=torch.float16
 ).to('cuda')
+
 pipeline.load_lora_weights(
-    'Prgckwb/trpfrog-sd3.5m-lora',
+    'Prgckwb/trpfrog-sd3.5m-lora', 
     weight_name='pytorch_lora_weights.safetensors'
 )
-image = pipeline('an icon of trpfrog eating ramen').images[0]
+image = pipeline('an icon of trpfrog').images[0]
+image.save('trpfrog.png')
 ```
 
 ### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
 
-- **LoRA**: download **[`diffusers_lora_weights.safetensors` here 💾](/Prgckwb/trpfrog-sd3.5m-lora/blob/main/pytorch_lora_weights.safetensors)**.
+- **LoRA**: download **[`pytorch_lora_weights.safetensors` here 💾](/Prgckwb/trpfrog-sd3.5-medium-lora/blob/main/pytorch_lora_weights.safetensors)**.
     - Rename it and place it on your `models/Lora` folder.
     - On AUTOMATIC1111, load the LoRA by adding `<lora:your_new_name:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
 
@@ -90,4 +92,3 @@ For more details, including weighting, merging and fusing LoRAs, check the [docu
 ## License
 
 Please adhere to the licensing terms as described [here](https://huggingface.co/stabilityai/stable-diffusion-3-medium/blob/main/LICENSE.md).
-