Update README.md

README.md

@@ -60,16 +60,28 @@ pip install diffusers transformers accelerate --upgrade
 
 - **Text-to-image**:
 
+To run the model yourself, you can leverage the 🧨 Diffusers library:
+1. Install the library:
+```
+pip install --upgrade diffusers  
+```
+
+2. Run the model:
 ```py
-from diffusers import stable_diffusion_custom_v4_1
+from TBA import CustomStableDiffusionPipeline4_1
 import torch
 
-pipe = AutoPipelineForText2Image.from_pretrained("vyroAI/ImagineV4.1", torch_dtype=torch.float16, variant="fp16")
-pipe.to("cuda")
+pipe = DiffusionPipeline.from_pretrained("vyroAI/ImagineV4.1")
+
+# To save GPU memory, torch.float16 can be used, but it may compromise image quality.
+pipe.to(torch_device="cuda", torch_dtype=torch.float16)
+
+prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"
 
-prompt = "A cinematic shot of a baby panda wearing an italian priest robe."
+# Can be set to 1~50 steps. 
+num_inference_steps = 30 
 
-image = pipe(prompt=prompt, num_inference_steps=30).images[0]
+images = pipe(prompt=prompt, num_inference_steps=num_inference_steps).images
 ```
 
 ### Out-of-Scope Use