prithivMLmods commited on
Commit
555598c
·
verified ·
1 Parent(s): 47bf539

update app

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -121,6 +121,9 @@ pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Light_restoration",
121
  pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Relight",
122
  weight_name="Qwen-Edit-Relight.safetensors",
123
  adapter_name="relight")
 
 
 
124
 
125
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
126
  MAX_SEED = np.iinfo(np.int32).max
@@ -168,6 +171,8 @@ def infer(
168
  pipe.set_adapters(["light-restoration"], adapter_weights=[1.0])
169
  elif lora_adapter == "Relight":
170
  pipe.set_adapters(["relight"], adapter_weights=[1.0])
 
 
171
 
172
  if randomize_seed:
173
  seed = random.randint(0, MAX_SEED)
@@ -233,7 +238,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
233
  with gr.Row():
234
  lora_adapter = gr.Dropdown(
235
  label="Choose Editing Style",
236
- choices=["Photo-to-Anime", "Multiple-Angles", "Light-Restoration", "Relight"],
237
  value="Photo-to-Anime"
238
  )
239
  with gr.Accordion("Advanced Settings", open=False, visible=False):
@@ -252,6 +257,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
252
  ["examples/6.jpg", "Switch the camera to a bottom-up view.", "Multiple-Angles"],
253
  ["examples/6.jpg", "Rotate the camera 180 degrees upside down.", "Multiple-Angles"],
254
  ["examples/4.jpg", "Rotate the camera 45 degrees to the right.", "Multiple-Angles"],
 
255
  ["examples/4.jpg", "Switch the camera to a top-down view.", "Multiple-Angles"],
256
  ["examples/4.jpg", "Switch the camera to a wide-angle lens.", "Multiple-Angles"],
257
  ],
 
121
  pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Relight",
122
  weight_name="Qwen-Edit-Relight.safetensors",
123
  adapter_name="relight")
124
+ pipe.load_lora_weights("eigen-ai-labs/eigen-banana-qwen-image-edit",
125
+ weight_name="eigen-banana-qwen-image-edit-fp16-lora.safetensors",
126
+ adapter_name="eigen-banana")
127
 
128
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
129
  MAX_SEED = np.iinfo(np.int32).max
 
171
  pipe.set_adapters(["light-restoration"], adapter_weights=[1.0])
172
  elif lora_adapter == "Relight":
173
  pipe.set_adapters(["relight"], adapter_weights=[1.0])
174
+ elif lora_adapter == "Eigen-Banana":
175
+ pipe.set_adapters(["eigen-banana"], adapter_weights=[1.0])
176
 
177
  if randomize_seed:
178
  seed = random.randint(0, MAX_SEED)
 
238
  with gr.Row():
239
  lora_adapter = gr.Dropdown(
240
  label="Choose Editing Style",
241
+ choices=["Photo-to-Anime", "Multiple-Angles", "Light-Restoration", "Relight", "Eigen-Banana"],
242
  value="Photo-to-Anime"
243
  )
244
  with gr.Accordion("Advanced Settings", open=False, visible=False):
 
257
  ["examples/6.jpg", "Switch the camera to a bottom-up view.", "Multiple-Angles"],
258
  ["examples/6.jpg", "Rotate the camera 180 degrees upside down.", "Multiple-Angles"],
259
  ["examples/4.jpg", "Rotate the camera 45 degrees to the right.", "Multiple-Angles"],
260
+ ["examples/7.jpg", "Apply a vintage film aesthetic to the image, featuring a subtle desaturation of colors with a warm, golden-hour tone, introduce a fine and natural-looking film grain across the entire scene, gently reduce overall contrast for a softer appearance, and add a very faint, dark vignette to the edges to mimic an aged photographic print.", "Eigen-Banana"],
261
  ["examples/4.jpg", "Switch the camera to a top-down view.", "Multiple-Angles"],
262
  ["examples/4.jpg", "Switch the camera to a wide-angle lens.", "Multiple-Angles"],
263
  ],