Update app.py

app.py

@@ -171,10 +171,12 @@ scheduler_config = {
 scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
 
 
-pipe = QwenImageEditInpaintPipeline.from_pretrained("Qwen/Qwen-Image-Edit", scheduler=scheduler, torch_dtype=torch.bfloat16).to("cuda")
+pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509", 
+                                                 scheduler=scheduler,
+                                                 torch_dtype=dtype).to(device)
 pipe.load_lora_weights(
         "lightx2v/Qwen-Image-Lightning", 
-        weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
+        weight_name="Qwen-Image-Edit-2509/Qwen-Image-Edit-2509-Lightning-8steps-V1.0-bf16.safetensors"
     )
 pipe.fuse_lora()
 
@@ -255,10 +257,10 @@ with gr.Blocks(css=css) as demo:
         """)
         gr.Markdown("""
         
-        Inpaint images with Qwen Image Edit. [Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series. 
+        Inpaint images with new [Qwen-Image-Edit-2509](https://huggingface.co/Qwen/Qwen-Image-Edit-2509). [Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series. 
         
         This demo uses the [Qwen-Image-Lightning](https://huggingface.co/lightx2v/Qwen-Image-Lightning) LoRA with FA3 for accelerated 8-step inference.
-        Try on [Qwen Chat](https://chat.qwen.ai/), or [download model](https://huggingface.co/Qwen/Qwen-Image-Edit) to run locally with ComfyUI or diffusers.
+        Try on [Qwen Chat](https://chat.qwen.ai/), or [download model](https://huggingface.co/Qwen/Qwen-Image-Edit-2509) to run locally with ComfyUI or diffusers.
         """)
         with gr.Row():
             with gr.Column():