update app

app.py

@@ -121,6 +121,10 @@ pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Fusion",
                        weight_name="溶图.safetensors", 
                        adapter_name="fusion-x")
 
+pipe.load_lora_weights("oumoumad/Qwen-Edit-2509-Material-transfer", 
+                       weight_name="material-transfer_000004769.safetensors", 
+                       adapter_name="material-transfer")
+
 try:
     pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
     print("Flash Attention 3 Processor set successfully.")
@@ -174,12 +178,15 @@ def infer(
             prompt = "Fuse object into background."
         elif lora_adapter == "Super-Fusion":
             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"
             
     adapters_map = {
         "Texture Edit": "texture",
         "Fuse-Objects": "fusion",
         "Cloth-Design-Fuse": "shirt_design",
         "Super-Fusion": "fusion-x",
+        "Material-Transfer": "material-transfer",
     }
     
     active_adapter = adapters_map.get(lora_adapter)
@@ -268,7 +275,7 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     lora_adapter = gr.Dropdown(
                         label="Choose Editing Style",
-                        choices=["Texture Edit", "Cloth-Design-Fuse", "Fuse-Objects", "Super-Fusion"],
+                        choices=["Texture Edit", "Cloth-Design-Fuse", "Fuse-Objects", "Super-Fusion", "Material-Transfer"],
                         value="Texture Edit",
                     )        
  
@@ -280,6 +287,7 @@ with gr.Blocks() as demo:
                 ["examples/F3.jpg", "examples/F4.jpg", "Replace her glasses with the new glasses from image 1.", "Super-Fusion"],
                 ["examples/F1.jpg", "examples/F2.jpg", "Put the small bottle on the table.", "Super-Fusion"],
                 ["examples/Mug1.jpg", "examples/Texture1.jpg", "Apply the design from image 2 to the mug.", "Texture Edit"],
+                ["examples/Chair.jpg", "examples/Material.jpg", "Change materials of image1 to match the reference in image2.", "Material-Transfer"],
                 ["examples/Cat1.jpg", "examples/Glass1.webp", "A cat wearing glasses in image 2.", "Fuse-Objects"],
 
             ],