Update app.py

app.py

@@ -216,7 +216,7 @@ def resize_to_fit(max_size, original_size):
  return new_width, new_height
 
  
-def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_type): 
+def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_type, , steps, control_strength): 
 
  max_size = (1500,1500)
  if prompt in prompt_list:
@@ -254,7 +254,10 @@ def process_generate(fore, prompt, intensity, mode, refprompt, isrmbg, model_typ
  "intensity" : float(intensity),
  "width" : 1500,
  "height" : 1500,
- "refprompt" : refprompt
+ "refprompt" : refprompt,
+ "first_stage_steps" : int(steps),
+ "first_stage_strength" : float(control_strength),
+ "second_stage_steps" : 30
  }
 
  # Select the endpoint based on the model type
@@ -520,6 +523,11 @@ with block:
 
  with gr.Column():
  model_type = gr.Dropdown(choices=["realistic","anime"], label="Model Type", value="realistic") # Adding the Dropdown for model selection
+
+ # Adding Advanced Settings for the number of steps and control strength
+ with gr.Accordion("Advanced Settings", open=False):
+ steps = gr.Slider(label="Number of Steps", minimum=10, maximum=100, value=30, step=1)
+ control_strength = gr.Slider(label="Control Strength", minimum=0.1, maximum=1.0, value=0.5, step=0.1)
  
  with gr.Column():
  gr.HTML('''
@@ -575,7 +583,7 @@ with block:
  
  
 
- ins = [fore, prompt, intensity, mode, refprompt, isrmbg, model_type]
+ ins = [fore, prompt, intensity, mode, refprompt, isrmbg, model_type, steps, control_strength]
  generate_button.click(fn=process_generate, inputs=ins, outputs=[result_gallery])