Update app.py
Browse files
app.py
CHANGED
|
@@ -123,7 +123,7 @@ with gr.Blocks() as demo:
|
|
| 123 |
max_steps = gr.Slider(label="Max Steps", minimum=50, maximum=200, value=200)
|
| 124 |
learning_rate = gr.Slider(label="Learning Rate", minimum=0.01, maximum=0.5, value=0.02)
|
| 125 |
optimization_steps = gr.Slider(label="Optimization Steps", minimum=1, maximum=10, value=1)
|
| 126 |
-
inverseproblem = gr.Checkbox(label="Apply mask on pixel space", value=False, info="Enables inverse problem formulation for inpainting by masking the RGB image itself. Hence, to avoid artifacts we increase the mask size manually during inference.")
|
| 127 |
mask_input = gr.Image(
|
| 128 |
type="pil",
|
| 129 |
label="Optional Mask",
|
|
@@ -268,7 +268,7 @@ with gr.Blocks() as demo:
|
|
| 268 |
200, # max_steps
|
| 269 |
0.02, # learning_rate
|
| 270 |
10, # optimization_steps
|
| 271 |
-
|
| 272 |
],
|
| 273 |
[
|
| 274 |
"./saved_results/20241129_212052/input.png", # image with mask
|
|
@@ -281,7 +281,7 @@ with gr.Blocks() as demo:
|
|
| 281 |
200, # max_steps
|
| 282 |
0.02, # learning_rate
|
| 283 |
10, # optimization_steps
|
| 284 |
-
|
| 285 |
],
|
| 286 |
[
|
| 287 |
"./saved_results/20241129_212155/input.png", # image with mask
|
|
@@ -294,7 +294,7 @@ with gr.Blocks() as demo:
|
|
| 294 |
200, # max_steps
|
| 295 |
0.02, # learning_rate
|
| 296 |
10, # optimization_steps
|
| 297 |
-
|
| 298 |
],
|
| 299 |
],
|
| 300 |
inputs=[
|
|
|
|
| 123 |
max_steps = gr.Slider(label="Max Steps", minimum=50, maximum=200, value=200)
|
| 124 |
learning_rate = gr.Slider(label="Learning Rate", minimum=0.01, maximum=0.5, value=0.02)
|
| 125 |
optimization_steps = gr.Slider(label="Optimization Steps", minimum=1, maximum=10, value=1)
|
| 126 |
+
inverseproblem = gr.Checkbox(label="Apply mask on pixel space (does not work well with HF ZeroGPU)", value=False, info="Enables inverse problem formulation for inpainting by masking the RGB image itself. Hence, to avoid artifacts we increase the mask size manually during inference.")
|
| 127 |
mask_input = gr.Image(
|
| 128 |
type="pil",
|
| 129 |
label="Optional Mask",
|
|
|
|
| 268 |
200, # max_steps
|
| 269 |
0.02, # learning_rate
|
| 270 |
10, # optimization_steps
|
| 271 |
+
False,
|
| 272 |
],
|
| 273 |
[
|
| 274 |
"./saved_results/20241129_212052/input.png", # image with mask
|
|
|
|
| 281 |
200, # max_steps
|
| 282 |
0.02, # learning_rate
|
| 283 |
10, # optimization_steps
|
| 284 |
+
False,
|
| 285 |
],
|
| 286 |
[
|
| 287 |
"./saved_results/20241129_212155/input.png", # image with mask
|
|
|
|
| 294 |
200, # max_steps
|
| 295 |
0.02, # learning_rate
|
| 296 |
10, # optimization_steps
|
| 297 |
+
False,
|
| 298 |
],
|
| 299 |
],
|
| 300 |
inputs=[
|