Update README.md

README.md

@@ -21,11 +21,30 @@ You can [use the diffusers library to load the model](https://huggingface.co/Pru
 To ensure that all optimizations are applied, use the pruna library to load the model using the following code:
 
 ```python
+import torch
+from diffusers import FluxFillPipeline
+from diffusers.utils import load_image
+
 from pruna import PrunaModel
 
-loaded_model = PrunaModel.from_hub(
+image = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/cup.png")
+mask = load_image("https://huggingface.co/datasets/diffusers/diffusers-images-docs/resolve/main/cup_mask.png")
+
+pipe = PrunaModel.from_hub(
     "PrunaAI/FLUX.1-Fill-dev-smashed"
 )
+image = pipe(
+    prompt="a white paper cup",
+    image=image,
+    mask_image=mask,
+    height=1632,
+    width=1232,
+    guidance_scale=30,
+    num_inference_steps=50,
+    max_sequence_length=512,
+    generator=torch.Generator("cpu").manual_seed(0)
+).images[0]
+image.save(f"flux-fill-dev.png")
 ```
 
 After loading the model, you can use the inference methods of the original model. Take a look at the [documentation](https://pruna.readthedocs.io/en/latest/index.html) for more usage information.