LPX55 commited on
Commit
4a06b0d
·
verified ·
1 Parent(s): 14e6c2f

Update app_local.py

Browse files
Files changed (1) hide show
  1. app_local.py +3 -2
app_local.py CHANGED
@@ -423,6 +423,7 @@ def infer(
423
  else:
424
  batch_prompts = [prompt] # Single prompt in list
425
  # Handle regular prompt rewriting
 
426
  if rewrite_prompt:
427
  try:
428
  enhanced_instruction = polish_prompt(original_prompt)
@@ -483,7 +484,7 @@ def infer(
483
  num_inference_steps=num_inference_steps,
484
  generator=generator,
485
  true_cfg_scale=varied_guidance,
486
- num_images_per_prompt=num_images_per_prompt
487
  ).images
488
  edited_images.extend(result)
489
  print(f"Generated image {i+1}/{len(batch_prompts)} with prompt: {current_prompt}...")
@@ -642,7 +643,7 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
642
  if preset_type and preset_type in ORIGINAL_PRESETS: # Changed from PRESETS to ORIGINAL_PRESETS
643
  non_empty_count = len([p for p in [prompt_1, prompt_2, prompt_3, prompt_4] if p.strip()])
644
  return gr.Slider(value=max(1, min(4, non_empty_count)), interactive=False)
645
- return gr.Slider(interactive=True)
646
 
647
  # Update the preset_dropdown.change handlers to use ORIGINAL_PRESETS
648
  preset_dropdown.change(
 
423
  else:
424
  batch_prompts = [prompt] # Single prompt in list
425
  # Handle regular prompt rewriting
426
+
427
  if rewrite_prompt:
428
  try:
429
  enhanced_instruction = polish_prompt(original_prompt)
 
484
  num_inference_steps=num_inference_steps,
485
  generator=generator,
486
  true_cfg_scale=varied_guidance,
487
+ num_images_per_prompt=2
488
  ).images
489
  edited_images.extend(result)
490
  print(f"Generated image {i+1}/{len(batch_prompts)} with prompt: {current_prompt}...")
 
643
  if preset_type and preset_type in ORIGINAL_PRESETS: # Changed from PRESETS to ORIGINAL_PRESETS
644
  non_empty_count = len([p for p in [prompt_1, prompt_2, prompt_3, prompt_4] if p.strip()])
645
  return gr.Slider(value=max(1, min(4, non_empty_count)), interactive=False)
646
+ return gr.Slider(interactive=True, visible=False)
647
 
648
  # Update the preset_dropdown.change handlers to use ORIGINAL_PRESETS
649
  preset_dropdown.change(