This PR allows to set the Number of inference steps

app.py

@@ -49,7 +49,7 @@ def clear_all():
     return gr.update(value=None), gr.update(value=None), gr.update(value=[], visible=False), gr.update(visible=False), gr.update(visible=False)
 
 @spaces.GPU()
-def generate(image_editor, prompt, neg_prompt, versions):
+def generate(image_editor, prompt, neg_prompt, versions, num_inference_steps):
     start = time.time()
     image = image_editor['background'].convert('RGB')
 
@@ -73,7 +73,8 @@ def generate(image_editor, prompt, neg_prompt, versions):
     pipeline.to("cuda")
     final_image = pipeline(prompt=prompt, 
                            image=image, 
-                           mask_image=mask).images[0]
+                           mask_image=mask,
+                           num_inference_steps=num_inference_steps).images[0]
 
 
     # Make sure the longest side of image is 1024
@@ -112,9 +113,10 @@ with gr.Blocks() as demo:
         with gr.Column():
             sketch_pad = gr.ImageMask(type='pil', label='Inpaint')
             prompt = gr.Textbox(label="Prompt")
-            generate_button = gr.Button("Generate")
+            generate_button = gr.Button(value="Inpaint", variant="primary")
             with gr.Accordion("Advanced Settings", open=False):
                 neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed')
+                num_inference_steps = gr.Slider(minimum = 10, maximum = 100, value = 30, step = 1, label = "Number of inference steps", info = "lower=faster, higher=image quality")
         with gr.Column():
             version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain', visible=False)
             restore_button = gr.Button("Restore Version", visible=False)
@@ -130,7 +132,7 @@ with gr.Blocks() as demo:
     # )
 
     version_gallery.select(get_select_index, None, selected)
-    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, version_gallery, restore_button, clear_button])
+    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery, num_inference_steps], outputs=[sketch_pad, version_gallery, restore_button, clear_button])
     restore_button.click(fn=restore_version, inputs=[selected, version_gallery], outputs=sketch_pad)
     clear_button.click(clear_all, inputs=None, outputs=[sketch_pad, prompt, version_gallery, restore_button, clear_button])