Update app.py

app.py

@@ -20,7 +20,7 @@ def update_textbox(selected_items):
     return ", ".join(selected_items)
 
 
-pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509", torch_dtype=torch.bfloat16)
+pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2511", torch_dtype=torch.bfloat16)
 print("pipeline loaded")
 
 pipe.to('cuda')
@@ -50,8 +50,8 @@ pipe = QwenImagePipeline.from_pretrained(
 
 
 
-qwenstyle= hf_hub_download(repo_id="Tele-AI/TeleStyleV2", filename="diffusers-TeleStyleV2-QIE-2509-Lora-bf16.safetensors")
-speedup = hf_hub_download(repo_id="Tele-AI/TeleStyleV2", filename="QIE-2509-Lightning-4steps-V1.0-bf16.safetensors")
+qwenstyle= hf_hub_download(repo_id="witcherderivia/PromptRoute/diffusers_promptroute_qie2511_bf16.safetensors", filename="diffusers-TeleStyleV2-QIE-2509-Lora-bf16.safetensors")
+speedup = hf_hub_download(repo_id="Tele-AI/TeleStyleV2", filename="QIE-2511-Lightning-4steps-V1.0-bf16.safetensors")
 
 
 
@@ -223,6 +223,21 @@ def infer(
             sw=minedge
             sh=int(sw*r)-int(sw*r)%16
 
+    if "Compose the pictures" in prompt:
+        
+        #minedge=1024
+        if content_ref:
+            w,h=content_ref.size
+            r=w/h
+            h=minedge
+            w=int(h*r)-int(h*r)%16
+        #content_ref=content_ref.resize((w,h))
+        #minedge=1024
+        if style_ref:
+            sw,sh=style_ref.size
+            r=sw/sh
+            sh=minedge
+            sw=int(sh*r)-int(sh*r)%16
 
     
     
@@ -258,6 +273,10 @@ def infer(
         prompt=prompt.replace(style_prompt.strip(),'')
     prompt=prompt.strip(',')
     print(f"Calling pipeline with prompt: '{prompt}'")
+    if "Compose the pictures" in prompt:
+            w=int(16/9*minedge)
+            w=w-w%16
+            h=minedge
     inputs = {
         "image": images,
         "prompt": prompt,