atatakun commited on
Commit
3d1c700
·
1 Parent(s): 2b42250

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -214,8 +214,8 @@ def color_shuffler(img, res):
214
  model_inpaint = None
215
 
216
 
217
- def inpaint(img, invert, res):
218
- image = resize_image(img, res)
219
  color = HWC3(image["image"])
220
  if(invert):
221
  alpha = image["mask"][:, :, 0:1]
@@ -252,7 +252,7 @@ with block:
252
  with gr.Column():
253
  gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
254
  # run_button.click(fn=inpaint, inputs=[input_image, resolution], outputs=[gallery])
255
- run_button.click(fn=inpaint, inputs=[input_image, invert, resolution], outputs=[gallery])
256
 
257
  gr.Markdown("<hr>")
258
  with gr.Row():
 
214
  model_inpaint = None
215
 
216
 
217
+ def inpaint(img, invert):
218
+ # image = resize_image(img, res)
219
  color = HWC3(image["image"])
220
  if(invert):
221
  alpha = image["mask"][:, :, 0:1]
 
252
  with gr.Column():
253
  gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
254
  # run_button.click(fn=inpaint, inputs=[input_image, resolution], outputs=[gallery])
255
+ run_button.click(fn=inpaint, inputs=[input_image, invert], outputs=[gallery])
256
 
257
  gr.Markdown("<hr>")
258
  with gr.Row():