Spaces:

editing-images
/

ledits

Running on A10G

Linoy Tsaban commited on May 27, 2023

Commit

5c65e56

•

1 Parent(s): 3db28ba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ def edit(input_image,
     pure_ddpm_out = sample(wt, zs, wts, prompt_tar=tar_prompt,
                            cfg_scale_tar=tar_cfg_scale, skip=skip,
                            eta = eta)
-    yield pure_ddpm_out
     editing_args = dict(
     editing_prompt = [edit_concept],

     pure_ddpm_out = sample(wt, zs, wts, prompt_tar=tar_prompt,
                            cfg_scale_tar=tar_cfg_scale, skip=skip,
                            eta = eta)
+    yield pure_ddpm_out, pure_ddpm_out
     editing_args = dict(
     editing_prompt = [edit_concept],