amildravid4292 commited on
Commit
0728566
·
verified ·
1 Parent(s): 32ac86d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -387,7 +387,7 @@ def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
387
  init_image = dict["background"].convert("RGB").resize((512, 512))
388
  mask = dict["layers"][0].convert("RGB").resize((512, 512))
389
 
390
- net = invert(init_image, mask, pcs, epochs, weight_decay,lr)
391
 
392
  #sample an image
393
  prompt = "sks person"
 
387
  init_image = dict["background"].convert("RGB").resize((512, 512))
388
  mask = dict["layers"][0].convert("RGB").resize((512, 512))
389
 
390
+ net = invert([init_image], mask, pcs, epochs, weight_decay,lr)
391
 
392
  #sample an image
393
  prompt = "sks person"