Update README.md

README.md

@@ -19,15 +19,22 @@ tags:
 import torch
 from diffusers import FluxKontextPipeline
 from diffusers.utils import load_image
+import uuid
 
 pipe = FluxKontextPipeline.from_pretrained("kpsss34/FLUX.1-Kontext-dev-int4", torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
-input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
+input_image = load_image("./1.jpg")
 
 image = pipe(
   image=input_image,
-  prompt="Add a hat to the cat",
-  guidance_scale=2.5
+  prompt="woman wearing dress",
+  guidance_scale=2.5,
+  num_inference_steps=30
 ).images[0]
+
+filename = f"generated_image_{uuid.uuid4().hex}.png"
+image.save(filename)
+
+print(f"Saved image as {filename}")
 ```