prithivMLmods commited on
Commit
de31997
·
verified ·
1 Parent(s): 488ea93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
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("eggnog/Qwen-Edit-2509-Stereogram_LoRa",
121
- weight_name="stereo.safetensors",
122
- adapter_name="stereogram")
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 == "Stereogram":
176
- prompt = "Make Stereogram."
177
 
178
  adapters_map = {
179
  "Texture Edit": "texture",
180
  "Fuse-Objects": "fusion",
181
  "Cloth-Design-Fuse": "shirt_design",
182
- "Stereogram": "stereogram",
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", "Stereogram"],
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/S1.jpg", "examples/S2.jpg", "Make Stereogram.", "Stereogram"],
 
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"],