update app

app.py

@@ -125,6 +125,10 @@ pipe.load_lora_weights("oumoumad/Qwen-Edit-2509-Material-transfer",
                        weight_name="material-transfer_000004769.safetensors", 
                        adapter_name="material-transfer")
 
+pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Light-Migration", 
+                       weight_name="参考色调.safetensors", 
+                       adapter_name="light-migration")
+
 try:
     pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
     print("Flash Attention 3 Processor set successfully.")
@@ -180,6 +184,8 @@ def infer(
             prompt = "Blend the product into the background, correct its perspective and lighting, and make it naturally integrated with the scene."
         elif lora_adapter == "Material-Transfer":
             prompt = "change materials of image1 to match the reference in image2"
+        elif lora_adapter == "Light-Migration":
+            prompt = "Refer to the color tone, remove the original lighting from Image 1, and relight Image 1 based on the lighting and color tone of Image 2."
             
     adapters_map = {
         "Texture Edit": "texture",
@@ -187,6 +193,7 @@ def infer(
         "Cloth-Design-Fuse": "shirt_design",
         "Super-Fusion": "fusion-x",
         "Material-Transfer": "material-transfer",
+        "Light-Migration": "light-migration",
     }
     
     active_adapter = adapters_map.get(lora_adapter)
@@ -281,6 +288,7 @@ with gr.Blocks() as demo:
  
         gr.Examples(
             examples=[
+                ["examples/M1.jpg", "examples/M2.jpg", "Refer to the color tone, remove the original lighting from Image 1, and relight Image 1 based on the lighting and color tone of Image 2.", "Light-Migration"],
                 ["examples/Cloth2.jpg", "examples/Design2.png", "Put this design on their shirt.", "Cloth-Design-Fuse"],
                 ["examples/Cup1.png", "examples/Wood1.png", "Apply wood texture to mug.", "Texture Edit"],
                 ["examples/Cloth1.jpg", "examples/Design1.png", "Put this design on their shirt.", "Cloth-Design-Fuse"],
@@ -305,4 +313,4 @@ with gr.Blocks() as demo:
     )
     
 if __name__ == "__main__":
-    demo.queue(max_size=30).launch(css=css, theme=orange_red_theme, mcp_server=True, ssr_mode=False, show_error=True)
+    demo.queue(max_size=50).launch(css=css, theme=orange_red_theme, mcp_server=True, ssr_mode=False, show_error=True)