prithivMLmods commited on
Commit
39e341b
·
verified ·
1 Parent(s): d67037d

update app

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -127,6 +127,9 @@ pipe.load_lora_weights("eigen-ai-labs/eigen-banana-qwen-image-edit",
127
  pipe.load_lora_weights("tlennon-ie/qwen-edit-skin",
128
  weight_name="qwen-edit-skin_1.1_000002750.safetensors",
129
  adapter_name="edit-skin")
 
 
 
130
 
131
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
132
  MAX_SEED = np.iinfo(np.int32).max
@@ -178,7 +181,9 @@ def infer(
178
  pipe.set_adapters(["eigen-banana"], adapter_weights=[1.0])
179
  elif lora_adapter == "Edit-Skin":
180
  pipe.set_adapters(["edit-skin"], adapter_weights=[1.0])
181
-
 
 
182
  if randomize_seed:
183
  seed = random.randint(0, MAX_SEED)
184
 
@@ -243,7 +248,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
243
  with gr.Row():
244
  lora_adapter = gr.Dropdown(
245
  label="Choose Editing Style",
246
- choices=["Photo-to-Anime", "Multiple-Angles", "Light-Restoration", "Relight", "Edit-Skin", "Eigen-Banana"],
247
  value="Photo-to-Anime"
248
  )
249
  with gr.Accordion("Advanced Settings", open=False, visible=False):
@@ -259,6 +264,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
259
  ["examples/4.jpg", "Use a subtle golden-hour filter with smooth light diffusion.", "Relight"],
260
  ["examples/2.jpeg", "Rotate the camera 45 degrees to the left.", "Multiple-Angles"],
261
  ["examples/2.jpeg", "Switch the camera to a top-down right corner view.", "Multiple-Angles"],
 
262
  ["examples/8.jpg", "Make the subjects skin details more prominent and natural.", "Edit-Skin"],
263
  ["examples/6.jpg", "Switch the camera to a bottom-up view.", "Multiple-Angles"],
264
  ["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"],
 
127
  pipe.load_lora_weights("tlennon-ie/qwen-edit-skin",
128
  weight_name="qwen-edit-skin_1.1_000002750.safetensors",
129
  adapter_name="edit-skin")
130
+ pipe.load_lora_weights("lovis93/next-scene-qwen-image-lora-2509",
131
+ weight_name="next-scene_lora-v2-3000.safetensors",
132
+ adapter_name="next-scene")
133
 
134
  pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
135
  MAX_SEED = np.iinfo(np.int32).max
 
181
  pipe.set_adapters(["eigen-banana"], adapter_weights=[1.0])
182
  elif lora_adapter == "Edit-Skin":
183
  pipe.set_adapters(["edit-skin"], adapter_weights=[1.0])
184
+ elif lora_adapter == "Next-Scene":
185
+ pipe.set_adapters(["next-scene"], adapter_weights=[1.0])
186
+
187
  if randomize_seed:
188
  seed = random.randint(0, MAX_SEED)
189
 
 
248
  with gr.Row():
249
  lora_adapter = gr.Dropdown(
250
  label="Choose Editing Style",
251
+ choices=["Photo-to-Anime", "Multiple-Angles", "Light-Restoration", "Relight", "Next-Scene", "Edit-Skin", "Eigen-Banana"],
252
  value="Photo-to-Anime"
253
  )
254
  with gr.Accordion("Advanced Settings", open=False, visible=False):
 
264
  ["examples/4.jpg", "Use a subtle golden-hour filter with smooth light diffusion.", "Relight"],
265
  ["examples/2.jpeg", "Rotate the camera 45 degrees to the left.", "Multiple-Angles"],
266
  ["examples/2.jpeg", "Switch the camera to a top-down right corner view.", "Multiple-Angles"],
267
+ ["examples/9.jpg", "The camera moves slightly forward as sunlight breaks through the clouds, casting a soft glow around the character's silhouette in the mist. Realistic cinematic style, atmospheric depth.", "Next-Scene"],
268
  ["examples/8.jpg", "Make the subjects skin details more prominent and natural.", "Edit-Skin"],
269
  ["examples/6.jpg", "Switch the camera to a bottom-up view.", "Multiple-Angles"],
270
  ["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"],