Update app.py

app.py

@@ -239,9 +239,19 @@ pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2511",
                                                  torch_dtype=dtype).to(device)
 pipe.load_lora_weights(
         "lightx2v/Qwen-Image-Edit-2511-Lightning", 
-        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
+        weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors",adapter_name="fast"
 )
-pipe.fuse_lora()
+pipe.load_lora_weights(
+        "lilylilith/AnyPose", 
+        weight_name="2511-AnyPose-base-000006250.safetensors",adapter_name="base"
+)
+pipe.load_lora_weights(
+        "lilylilith/AnyPose", 
+        weight_name="2511-AnyPose-helper-00006000.safetensors",adapter_name="helper"
+)
+pipe.set_adapters(["fast", "base", "helper"], adapter_weights=[1., 0.7, 0.7])
+pipe.fuse_lora(adapter_names=["fast", "base", "helper"], lora_scale=1)
+pipe.unload_lora_weights()
 
 # # Apply the same optimizations from the first version
 # pipe.transformer.__class__ = QwenImageTransformer2DModel
@@ -253,6 +263,15 @@ pipe.fuse_lora()
 # --- UI Constants and Helpers ---
 MAX_SEED = np.iinfo(np.int32).max
 
+DEFAULT_LORA_PROMPT = """
+Make the person in image 1 do the exact same pose of the person in image 2. 
+Changing the style and background of the image of the person in image 1 is undesirable, so don't do it. 
+The new pose should be pixel accurate to the pose we are trying to copy. 
+The position of the arms and head and legs should be the same as the pose we are trying to copy. 
+Change the field of view and angle to match exactly image 2. Head tilt and eye gaze pose should match the person in image 2. 
+Remove the background of image 2, and replace it with the background of image 1.
+"""
+
 def use_output_as_input(output_images):
     """Convert output images to input format for the gallery"""
     if output_images is None or len(output_images) == 0:
@@ -388,11 +407,11 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             prompt = gr.Text(
                     label="Prompt",
+                    value=DEFAULT_LORA_PROMPT,
                     show_label=False,
-                    placeholder="describe the edit instruction",
-                    container=False,
+                    visible=False
             )
-            run_button = gr.Button("Edit!", variant="primary")
+            run_button = gr.Button("Edit Pose", variant="primary")
 
         with gr.Accordion("Advanced Settings", open=False):
             # Negative prompt UI element is removed here
@@ -442,7 +461,7 @@ with gr.Blocks(css=css) as demo:
                 )
                 
                 
-                rewrite_prompt = gr.Checkbox(label="Rewrite prompt", value=True)
+                rewrite_prompt = gr.Checkbox(label="Rewrite prompt", value=False, visible=False)
 
         # gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False)