Update README.md

README.md

@@ -12,7 +12,9 @@ tags:
 
 # Shuttle 3 Diffusion
 
-Shuttle 3 Diffusion is a text-to-image AI model designed to create detailed and diverse images from textual prompts. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
+Shuttle 3 Diffusion is a text-to-image AI model designed to create detailed and diverse images from textual prompts in just 4 steps. It offers enhanced performance in image quality, typography, understanding complex prompts, and resource efficiency.
+
+![image/png](https://huggingface.co/shuttleai/shuttle-3-diffusion/resolve/main/demo.png)
 
 You can try out the model through a website at https://chat.shuttleai.com/images
 
@@ -66,13 +68,14 @@ image.save("shuttle.png")
 ```
 To learn more check out the [diffusers](https://huggingface.co/docs/diffusers/main/en/api/pipelines/flux) documentation
 
-## Using the model with Automatic1111/ComfyUI
+## Using the model with ComfyUI
 
-Support coming soon. We will update this model card with instructions when ready.
+To run local inference with Shuttle 3 Diffusion using [ComfyUI](https://github.com/comfyanonymous/ComfyUI), you can use this [safetensors file](https://huggingface.co/shuttleai/shuttle-3-diffusion/blob/main/shuttle-3-diffusion.safetensors).
 
 ## Comparison to other models
-Shuttle 3 Diffusion can produce better images than Flux Dev in just four steps, while being licensed under Apache 2.
+Shuttle 3 Diffusion can produce images better images than Flux Dev in just four steps, while being licensed under Apache 2.
 ![image/png](https://huggingface.co/shuttleai/shuttle-3-diffusion/resolve/main/comparison.png)
+[More examples](https://docs.shuttleai.com/getting-started/shuttle-diffusion)
 
 ## Training Details
-Shuttle 3 Diffusion uses Flux.1 Schnell as its base. It can produce images similar to Flux Pro in just 4 steps, and it is licensed under Apache 2. The model was partially de-distilled during training. When used beyond 10 steps, it enters "refiner mode," enhancing image details without altering the composition. We overcame the limitations of the Schnell-series models by employing a special training method, resulting in improved details and colors.
+Shuttle 3 Diffusion uses Flux.1 Schnell as its base. It can produce images similar to Flux Dev or Pro in just 4 steps, and it is licensed under Apache 2. The model was partially de-distilled during training. When used beyond 10 steps, it enters "refiner mode," enhancing image details without altering the composition. We overcame the limitations of the Schnell-series models by employing a special training method, resulting in improved details and colors.