parameterize guidance scale and blur mask

app.py

@@ -30,6 +30,7 @@ This Space appears to be running on a CPU; it will take hours to get results.  Y
 
 xlp_kwargs['torch_dtype'] = device_dtype
 
+
 def merge_images(original, new_image, offset, direction):
     if direction in ["left", "right"]:
         merged_image = np.zeros(
@@ -50,7 +51,7 @@ def merge_images(original, new_image, offset, direction):
         merged_image[: new_image.shape[0], :] = new_image
     elif direction == "bottom":
         merged_image[: original.shape[0], :] = original
-        merged_image[original.shape[0] + offset - new_image.shape[0]: original.shape[0] + offset, :] = new_image
+        merged_image[original.shape[0] + offset - new_image.shape[0]:original.shape[0] + offset, :] = new_image
 
     return merged_image
 
@@ -182,8 +183,9 @@ def image_resize(image, new_size=1024):
 
     return image
 
+
 @spaces.GPU
-def outpaint(pil_image, direction='right', times_to_expand=4):
+def outpaint(pil_image, direction='right', times_to_expand=4, guidance_scale=4.0, blur_radius=500):
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
 
@@ -215,7 +217,7 @@ def outpaint(pil_image, direction='right', times_to_expand=4):
             negative_prompt=negative_prompt,
             width=1024,
             height=1024,
-            guidance_scale=4.0,
+            guidance_scale=guidance_scale,
             num_inference_steps=25,
             original_image=image,
             image=image,
@@ -231,7 +233,6 @@ def outpaint(pil_image, direction='right', times_to_expand=4):
 
         return image
 
-
     prompt = ""
     negative_prompt = ""
     inpaint_mask_color = 50  # lighter use more of the Telea inpainting
@@ -243,7 +244,7 @@ def outpaint(pil_image, direction='right', times_to_expand=4):
     # image.shape[1] for horizontal, image.shape[0] for vertical
     expand_pixels_to_square = 1024 - image.shape[1]
     image, mask = process_image(
-        image, expand_pixels=expand_pixels_to_square, direction=direction, inpaint_mask_color=inpaint_mask_color
+        image, expand_pixels=expand_pixels_to_square, direction=direction, inpaint_mask_color=inpaint_mask_color, blur_radius=blur_radius
     )
 
     ip_adapter_image = []
@@ -256,7 +257,7 @@ def outpaint(pil_image, direction='right', times_to_expand=4):
 
     for i in range(times_to_expand):
         image, mask = process_image(
-            final_image, direction=direction, expand_pixels=expand_pixels, inpaint_mask_color=inpaint_mask_color
+            final_image, direction=direction, expand_pixels=expand_pixels, inpaint_mask_color=inpaint_mask_color, blur_radius=blur_radius
         )
 
         ip_adapter_image = []
@@ -280,6 +281,10 @@ gradio_app = gr.Interface(
                  info="Outward from which edge to paint?", value='right'),
         gr.Slider(2, 4, step=1, value=4, label="Times to expand",
                   info="Choose between 2 and 4"),
+        gr.Slider(1, 12, step=0.1, value=4, label="Guidance scale",
+                  info="Choose between 1 and 12"),
+        gr.Slider(250, 500, step=1, value=500, label="Mask blur radius",
+                  info="Choose between 250 and 500"),
     ],
     outputs=[gr.Image(label="Processed Image")],
     title="Outpainting with differential diffusion demo",