Update app.py

app.py

@@ -21,10 +21,10 @@ pipe.to(device)
 
 gallery = gr.Gallery(label="Generated image", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
 
-def infer(prompt): 
+def infer(prompt, steps=21): 
     
     #image = pipe(prompt, init_image=init_image)["sample"][0]
-    images_list = pipe([prompt], num_inference_steps=21)
+    images_list = pipe([prompt], num_inference_steps=steps)
     images = []
     safe_image = Image.open(r"unsafe.png")
     for i, image in enumerate(images_list["images"]):