mpatel57 commited on
Commit
5a3565c
1 Parent(s): be24a5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- True,
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
- True,
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
- True,
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=[