amildravid4292 commited on
Commit
a76a3b6
·
verified ·
1 Parent(s): dc4a959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -373,7 +373,9 @@ def invert(self, image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e
373
 
374
 
375
  @spaces.GPU
376
- def run_inversion(self, dict, pcs, epochs, weight_decay,lr):
 
 
377
  init_image = dict["image"].convert("RGB").resize((512, 512))
378
  mask = dict["mask"].convert("RGB").resize((512, 512))
379
  network = invert([init_image], mask, pcs, epochs, weight_decay,lr)
@@ -501,9 +503,9 @@ with gr.Blocks(css="style.css") as demo:
501
 
502
 
503
 
504
- # invert_button.click(fn=model.run_inversion,
505
- # inputs=[input_image, pcs, epochs, weight_decay,lr],
506
- # outputs = [input_image, file_output])
507
 
508
 
509
  sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image])
 
373
 
374
 
375
  @spaces.GPU
376
+ def run_inversion(dict, pcs, epochs, weight_decay,lr):
377
+ print(dict)
378
+ print(dict.keys())
379
  init_image = dict["image"].convert("RGB").resize((512, 512))
380
  mask = dict["mask"].convert("RGB").resize((512, 512))
381
  network = invert([init_image], mask, pcs, epochs, weight_decay,lr)
 
503
 
504
 
505
 
506
+ invert_button.click(fn=run_inversion,
507
+ inputs=[input_image, pcs, epochs, weight_decay,lr],
508
+ outputs = [input_image, file_output])
509
 
510
 
511
  sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image])