Update app.py

app.py

@@ -119,7 +119,7 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
     return background, mask
 
 @spaces.GPU(duration=200)
-def inpaint(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, progress=gr.Progress(track_tqdm=True)):
+def inpaint(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom, guidance_scale, progress=gr.Progress(track_tqdm=True)):
     
     background, mask = prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom)
     
@@ -140,7 +140,7 @@ def inpaint(image, width, height, overlap_percentage, num_inference_steps, resiz
         image=cnet_image,
         mask_image=mask,
         num_inference_steps=num_inference_steps,
-        guidance_scale=30,
+        guidance_scale=guidance_scale,
     ).images[0]
    
     result = result.convert("RGBA")
@@ -238,6 +238,14 @@ with gr.Blocks(css=css) as demo:
                 with gr.Row():
                     with gr.Column(scale=2):
                         prompt_input = gr.Textbox(label="Prompt (Optional)")
+                    with gr.Column(scale=2):
+                        guidance_scale = gr.Slider(
+                            minimum=1,
+                            maximum=50,
+                            value=30,
+                            step=0.5,
+                            label="Guidance Scale"
+                        )
                     with gr.Column(scale=1):
                         run_button = gr.Button("Generate")
 
@@ -308,22 +316,23 @@ with gr.Blocks(css=css) as demo:
                     interactive=False,
                     label="Generated Image",
                 )
-                result_mask = gr.Image(
-                    interactive=False,
-                    label="Generated Mask",
-                )
-                result_pure = gr.Image(
-                    interactive=False,
-                    label="Generated Result",
-                )
-                cnet_pure = gr.Image(
-                    interactive=False,
-                    label="Cnet Pure",
-                )
-                background = gr.Image(
-                    interactive=False,
-                    label="background",
-                )
+                with gr.Accordion("Step Results", open=False):
+                    result_mask = gr.Image(
+                        interactive=False,
+                        label="Generated Mask",
+                    )
+                    result_pure = gr.Image(
+                        interactive=False,
+                        label="Generated Result",
+                    )
+                    cnet_pure = gr.Image(
+                        interactive=False,
+                        label="Cnet Pure",
+                    )
+                    background = gr.Image(
+                        interactive=False,
+                        label="background",
+                    )
                 use_as_input_button = gr.Button("Use as Input Image", visible=False)
                 with gr.Accordion("History and Mask", open=False):
                     history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
@@ -374,7 +383,7 @@ with gr.Blocks(css=css) as demo:
         fn=inpaint,
         inputs=[input_image, width_slider, height_slider, overlap_percentage, num_inference_steps,
                 resize_option, custom_resize_percentage, prompt_input, alignment_dropdown,
-                overlap_left, overlap_right, overlap_top, overlap_bottom],
+                overlap_left, overlap_right, overlap_top, overlap_bottom, guidance_scale],
         outputs=[result, preview_image, result_mask, result_pure, cnet_pure, background],
     ).then(
         fn=lambda x, history: update_history(x, history),