Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -117,9 +117,9 @@ pipe.load_lora_weights("ostris/qwen_image_edit_2509_shirt_design",
|
|
| 117 |
weight_name="qwen_image_edit_2509_shirt_design.safetensors",
|
| 118 |
adapter_name="shirt_design")
|
| 119 |
|
| 120 |
-
pipe.load_lora_weights("
|
| 121 |
-
weight_name="
|
| 122 |
-
adapter_name="
|
| 123 |
|
| 124 |
try:
|
| 125 |
pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
|
|
@@ -172,14 +172,14 @@ def infer(
|
|
| 172 |
prompt = "Apply texture to object."
|
| 173 |
elif lora_adapter == "Fuse-Objects":
|
| 174 |
prompt = "Fuse object into background."
|
| 175 |
-
elif lora_adapter == "
|
| 176 |
-
prompt = "
|
| 177 |
|
| 178 |
adapters_map = {
|
| 179 |
"Texture Edit": "texture",
|
| 180 |
"Fuse-Objects": "fusion",
|
| 181 |
"Cloth-Design-Fuse": "shirt_design",
|
| 182 |
-
"
|
| 183 |
}
|
| 184 |
|
| 185 |
active_adapter = adapters_map.get(lora_adapter)
|
|
@@ -269,7 +269,7 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
|
|
| 269 |
with gr.Row():
|
| 270 |
lora_adapter = gr.Dropdown(
|
| 271 |
label="Choose Editing Style",
|
| 272 |
-
choices=["Texture Edit", "Fuse-Objects", "Cloth-Design-Fuse"
|
| 273 |
value="Texture Edit",
|
| 274 |
)
|
| 275 |
|
|
@@ -277,7 +277,8 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
|
|
| 277 |
examples=[
|
| 278 |
["examples/Cloth2.jpg", "examples/Design2.png", "Put this design on their shirt.", "Cloth-Design-Fuse"],
|
| 279 |
["examples/Cup1.png", "examples/Wood1.png", "Apply wood texture to mug.", "Texture Edit"],
|
| 280 |
-
["examples/
|
|
|
|
| 281 |
["examples/Mug1.jpg", "examples/Texture1.jpg", "Apply the design from image 2 to the mug.", "Texture Edit"],
|
| 282 |
["examples/Cat1.jpg", "examples/Glass1.webp", "A cat wearing glasses in image 2.", "Fuse-Objects"],
|
| 283 |
["examples/Cloth1.jpg", "examples/Design1.png", "Put this design on their shirt.", "Cloth-Design-Fuse"],
|
|
|
|
| 117 |
weight_name="qwen_image_edit_2509_shirt_design.safetensors",
|
| 118 |
adapter_name="shirt_design")
|
| 119 |
|
| 120 |
+
pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Fusion",
|
| 121 |
+
weight_name="溶图.safetensors",
|
| 122 |
+
adapter_name="fusion")
|
| 123 |
|
| 124 |
try:
|
| 125 |
pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
|
|
|
|
| 172 |
prompt = "Apply texture to object."
|
| 173 |
elif lora_adapter == "Fuse-Objects":
|
| 174 |
prompt = "Fuse object into background."
|
| 175 |
+
elif lora_adapter == "Super-Fusion":
|
| 176 |
+
prompt = "Blend the product into the background, correct its perspective and lighting, and make it naturally integrated with the scene."
|
| 177 |
|
| 178 |
adapters_map = {
|
| 179 |
"Texture Edit": "texture",
|
| 180 |
"Fuse-Objects": "fusion",
|
| 181 |
"Cloth-Design-Fuse": "shirt_design",
|
| 182 |
+
"Super-Fusion": "fusion",
|
| 183 |
}
|
| 184 |
|
| 185 |
active_adapter = adapters_map.get(lora_adapter)
|
|
|
|
| 269 |
with gr.Row():
|
| 270 |
lora_adapter = gr.Dropdown(
|
| 271 |
label="Choose Editing Style",
|
| 272 |
+
choices=["Texture Edit", "Fuse-Objects", "Super-Fusion", "Cloth-Design-Fuse"],
|
| 273 |
value="Texture Edit",
|
| 274 |
)
|
| 275 |
|
|
|
|
| 277 |
examples=[
|
| 278 |
["examples/Cloth2.jpg", "examples/Design2.png", "Put this design on their shirt.", "Cloth-Design-Fuse"],
|
| 279 |
["examples/Cup1.png", "examples/Wood1.png", "Apply wood texture to mug.", "Texture Edit"],
|
| 280 |
+
["examples/F1.jpg", "examples/F2.jpg", "Put the small bottle on the table.", "Super-Fusion"],
|
| 281 |
+
["examples/F3.jpg", "examples/F4.jpg", "Replace her glasses.", "Super-Fusion"],
|
| 282 |
["examples/Mug1.jpg", "examples/Texture1.jpg", "Apply the design from image 2 to the mug.", "Texture Edit"],
|
| 283 |
["examples/Cat1.jpg", "examples/Glass1.webp", "A cat wearing glasses in image 2.", "Fuse-Objects"],
|
| 284 |
["examples/Cloth1.jpg", "examples/Design1.png", "Put this design on their shirt.", "Cloth-Design-Fuse"],
|