prithivMLmods commited on
Commit
d8d42ce
·
verified ·
1 Parent(s): 05011af

update app

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -133,6 +133,10 @@ pipe.load_lora_weights("lovis93/next-scene-qwen-image-lora-2509",
133
  pipe.load_lora_weights("vafipas663/Qwen-Edit-2509-Upscale-LoRA",
134
  weight_name="qwen-edit-enhance_64-v3_000001000.safetensors",
135
  adapter_name="upscale-image")
 
 
 
 
136
 
137
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
138
  MAX_SEED = np.iinfo(np.int32).max
@@ -188,6 +192,8 @@ def infer(
188
  pipe.set_adapters(["next-scene"], adapter_weights=[1.0])
189
  elif lora_adapter == "Upscale-Image":
190
  pipe.set_adapters(["upscale-image"], adapter_weights=[1.0])
 
 
191
 
192
  if randomize_seed:
193
  seed = random.randint(0, MAX_SEED)
@@ -253,7 +259,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
253
  with gr.Row():
254
  lora_adapter = gr.Dropdown(
255
  label="Choose Editing Style",
256
- choices=["Photo-to-Anime", "Multiple-Angles", "Light-Restoration", "Multi-Angle-Lighting", "Upscale-Image", "Relight", "Next-Scene", "Edit-Skin"],
257
  value="Photo-to-Anime"
258
  )
259
  with gr.Accordion("Advanced Settings", open=False, visible=False):
@@ -269,6 +275,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
269
  ["examples/4.jpg", "Use a subtle golden-hour filter with smooth light diffusion.", "Relight"],
270
  ["examples/2.jpeg", "Rotate the camera 45 degrees to the left.", "Multiple-Angles"],
271
  ["examples/7.jpg", "Light source from the Right Rear", "Multi-Angle-Lighting"],
 
272
  ["examples/10.jpeg", "Upscale the image.", "Upscale-Image"],
273
  ["examples/7.jpg", "Light source from the Below", "Multi-Angle-Lighting"],
274
  ["examples/2.jpeg", "Switch the camera to a top-down right corner view.", "Multiple-Angles"],
 
133
  pipe.load_lora_weights("vafipas663/Qwen-Edit-2509-Upscale-LoRA",
134
  weight_name="qwen-edit-enhance_64-v3_000001000.safetensors",
135
  adapter_name="upscale-image")
136
+ pipe.load_lora_weights("tarn59/extract_texture_qwen_image_edit_2509",
137
+ weight_name="extract_texture_qwen_image_edit_2509.safetensors",
138
+ adapter_name="extract-texture")
139
+
140
 
141
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
142
  MAX_SEED = np.iinfo(np.int32).max
 
192
  pipe.set_adapters(["next-scene"], adapter_weights=[1.0])
193
  elif lora_adapter == "Upscale-Image":
194
  pipe.set_adapters(["upscale-image"], adapter_weights=[1.0])
195
+ elif lora_adapter == "Extract-Texture":
196
+ pipe.set_adapters(["extract-texture"], adapter_weights=[1.0])
197
 
198
  if randomize_seed:
199
  seed = random.randint(0, MAX_SEED)
 
259
  with gr.Row():
260
  lora_adapter = gr.Dropdown(
261
  label="Choose Editing Style",
262
+ choices=["Photo-to-Anime", "Multiple-Angles", "Light-Restoration", "Multi-Angle-Lighting", "Upscale-Image", "Relight", "Next-Scene", "Edit-Skin", "Extract-Texture"],
263
  value="Photo-to-Anime"
264
  )
265
  with gr.Accordion("Advanced Settings", open=False, visible=False):
 
275
  ["examples/4.jpg", "Use a subtle golden-hour filter with smooth light diffusion.", "Relight"],
276
  ["examples/2.jpeg", "Rotate the camera 45 degrees to the left.", "Multiple-Angles"],
277
  ["examples/7.jpg", "Light source from the Right Rear", "Multi-Angle-Lighting"],
278
+ ["examples/11.jpeg", "Extract stone texture from the wall. Extract into a texture image.", "Extract-Texture"],
279
  ["examples/10.jpeg", "Upscale the image.", "Upscale-Image"],
280
  ["examples/7.jpg", "Light source from the Below", "Multi-Angle-Lighting"],
281
  ["examples/2.jpeg", "Switch the camera to a top-down right corner view.", "Multiple-Angles"],