Linoy Tsaban
commited on
Commit
·
f02e93c
1
Parent(s):
6a5a59b
Update app.py
Browse files
app.py
CHANGED
|
@@ -125,12 +125,12 @@ def reconstruct(tar_prompt,
|
|
| 125 |
wts, zs,
|
| 126 |
# do_reconstruction,
|
| 127 |
# reconstruction
|
| 128 |
-
)
|
| 129 |
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
|
| 135 |
def load_and_invert(
|
| 136 |
input_image,
|
|
|
|
| 125 |
wts, zs,
|
| 126 |
# do_reconstruction,
|
| 127 |
# reconstruction
|
| 128 |
+
):
|
| 129 |
|
| 130 |
+
|
| 131 |
+
# if do_reconstruction:
|
| 132 |
+
reconstruction = sample(zs.value, wts.value, prompt_tar=tar_prompt, skip=skip, cfg_scale_tar=tar_cfg_scale)
|
| 133 |
+
return reconstruction
|
| 134 |
|
| 135 |
def load_and_invert(
|
| 136 |
input_image,
|