Spaces:
Runtime error
Runtime error
add mask blur feature
Browse files
app.py
CHANGED
|
@@ -112,6 +112,8 @@ def process(
|
|
| 112 |
input_image_editor: dict,
|
| 113 |
image_url: str,
|
| 114 |
mask_url: str,
|
|
|
|
|
|
|
| 115 |
lora_path: str,
|
| 116 |
lora_weights: str,
|
| 117 |
lora_scale: float,
|
|
@@ -152,6 +154,8 @@ def process(
|
|
| 152 |
if not mask:
|
| 153 |
gr.Info("Please draw a mask on the image.")
|
| 154 |
return None, None
|
|
|
|
|
|
|
| 155 |
|
| 156 |
with calculateDuration("resize image"):
|
| 157 |
width, height = resize_image_dimensions(original_resolution_wh=image.size)
|
|
@@ -268,6 +272,15 @@ with gr.Blocks() as demo:
|
|
| 268 |
randomize_seed_checkbox_component = gr.Checkbox(
|
| 269 |
label="Randomize seed", value=True)
|
| 270 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 271 |
with gr.Row():
|
| 272 |
strength_slider_component = gr.Slider(
|
| 273 |
label="Strength",
|
|
@@ -302,6 +315,8 @@ with gr.Blocks() as demo:
|
|
| 302 |
input_image_editor_component,
|
| 303 |
image_url,
|
| 304 |
mask_url,
|
|
|
|
|
|
|
| 305 |
lora_path,
|
| 306 |
lora_weights,
|
| 307 |
lora_scale,
|
|
|
|
| 112 |
input_image_editor: dict,
|
| 113 |
image_url: str,
|
| 114 |
mask_url: str,
|
| 115 |
+
blur_mask: bool,
|
| 116 |
+
blur_factor: int,
|
| 117 |
lora_path: str,
|
| 118 |
lora_weights: str,
|
| 119 |
lora_scale: float,
|
|
|
|
| 154 |
if not mask:
|
| 155 |
gr.Info("Please draw a mask on the image.")
|
| 156 |
return None, None
|
| 157 |
+
if blur_mask:
|
| 158 |
+
mask = pipe.mask_processor.blur(mask, blur_factor=blur_factor)
|
| 159 |
|
| 160 |
with calculateDuration("resize image"):
|
| 161 |
width, height = resize_image_dimensions(original_resolution_wh=image.size)
|
|
|
|
| 272 |
randomize_seed_checkbox_component = gr.Checkbox(
|
| 273 |
label="Randomize seed", value=True)
|
| 274 |
|
| 275 |
+
blur_mask = gr.Checkbox(
|
| 276 |
+
label="if blur mask", value=False)
|
| 277 |
+
blur_factor = gr.Slider(
|
| 278 |
+
label="blur factor",
|
| 279 |
+
minimum=0,
|
| 280 |
+
maximum=50,
|
| 281 |
+
step=1,
|
| 282 |
+
value=33,
|
| 283 |
+
)
|
| 284 |
with gr.Row():
|
| 285 |
strength_slider_component = gr.Slider(
|
| 286 |
label="Strength",
|
|
|
|
| 315 |
input_image_editor_component,
|
| 316 |
image_url,
|
| 317 |
mask_url,
|
| 318 |
+
blur_mask,
|
| 319 |
+
blur_factor,
|
| 320 |
lora_path,
|
| 321 |
lora_weights,
|
| 322 |
lora_scale,
|