Update app.py

app.py

@@ -7,6 +7,7 @@ import os
 
 # from diffusers import QwenImageEditInpaintPipeline
 from optimization import optimize_pipeline_
+from diffusers.utils import load_image
 from qwenimage.pipeline_qwenimage_edit_inpaint import QwenImageEditInpaintPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
 from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
@@ -24,8 +25,10 @@ pipe = QwenImageEditInpaintPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torc
 pipe.transformer.__class__ = QwenImageTransformer2DModel
 pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 
+dummy_mask  = load_image("https://github.com/Trgtuan10/Image_storage/blob/main/mask_cat.png?raw=true")
+
 # --- Ahead-of-time compilation ---
-optimize_pipeline_(pipe, image=Image.new("RGB", (1024, 1024)), prompt="prompt")
+optimize_pipeline_(pipe, image=Image.new("RGB", (1328, 1328)), prompt="prompt", mask_image=dummy_mask)
 
 @spaces.GPU(duration=120)
 def infer(edit_images, prompt, negative_prompt="", seed=42, randomize_seed=False, strength=1.0, num_inference_steps=35, true_cfg_scale=4.0, progress=gr.Progress(track_tqdm=True)):