Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import numpy as np
 import random
@@ -13,17 +14,21 @@ from huggingface_hub import  hf_hub_download
 
 
 
-
 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", 
+        download_source='huggingface',
+        origin_file_pattern="transformer/diffusion_pytorch_model*.safetensors"),
+        ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", 
+        download_source='huggingface',origin_file_pattern="text_encoder/model*.safetensors"),
+        ModelConfig(model_id="Qwen/Qwen-Image-Edit-2509", 
+        download_source='huggingface',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", 
+    download_source='huggingface',origin_file_pattern="processor/"),
 )
 
 
@@ -126,7 +131,6 @@ _HEADER_ = '''
 
 
 <p style="font-size: 1rem; margin-bottom: 1.5rem;">Paper: <a href='https://openreview.net/forum?id=Cgb7JpOA5Q&referrer=%5Bthe%20profile%20of%20Shiwen%20Zhang%5D(%2Fprofile%3Fid%3D~Shiwen_Zhang1)' target='_blank'>QwenStyle: Content-Preserving Style Transfer with Qwen-Image-Edit</a> | Codes: <a href='https://github.com/witcherofresearch/Qwen-Image-Style-Transfer' target='_blank'>GitHub</a></p>
-<p style="font-size: 1rem; margin-bottom: 1.5rem;">If you encounter an Error with this demo, the most possible reason is ZeroGPU out-of-memory and the solution is to decrease the Min Edge of the generated image from 1024 to a lower value. This is because ZeroGPU has a memory limit of 70GB, while all the examples are tested with 80GB H100 GPUs.  </p>
 '''  
 
 with gr.Blocks() as demo: