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])