Update app.py
Browse files
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"
|