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,
|