Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import numpy as np
 import random
@@ -11,19 +12,24 @@ import os
 
 from huggingface_hub import  hf_hub_download
 
+from diffsynth import download_customized_models
 
-
+download_customized_models(
+    model_id="Qwen/Qwen-Image-Edit-2509",
+    local_dir="models/Qwen/Qwen-Image-Edit-2509",
+    downloading_priority=["HuggingFace"]
+)
 
 pipe = QwenImagePipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
     device="cuda",
     model_configs=[
         ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="transformer/diffusion_pytorch_model*.safetensors"),
-        ModelConfig(model_id="Qwen/Qwen-Image", origin_file_pattern="text_encoder/model*.safetensors"),
-        ModelConfig(model_id="Qwen/Qwen-Image", origin_file_pattern="vae/diffusion_pytorch_model.safetensors"),
+        ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="text_encoder/model*.safetensors"),
+        ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="vae/diffusion_pytorch_model.safetensors"),
     ],
     tokenizer_config=None,
-    processor_config=ModelConfig(model_id="Qwen/Qwen-Image-Edit", origin_file_pattern="processor/"),
+    processor_config=ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", origin_file_pattern="processor/"),
 )