Linoy Tsaban commited on
Commit
5c65e56
1 Parent(s): 3db28ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def edit(input_image,
77
  pure_ddpm_out = sample(wt, zs, wts, prompt_tar=tar_prompt,
78
  cfg_scale_tar=tar_cfg_scale, skip=skip,
79
  eta = eta)
80
- yield pure_ddpm_out
81
 
82
  editing_args = dict(
83
  editing_prompt = [edit_concept],
 
77
  pure_ddpm_out = sample(wt, zs, wts, prompt_tar=tar_prompt,
78
  cfg_scale_tar=tar_cfg_scale, skip=skip,
79
  eta = eta)
80
+ yield pure_ddpm_out, pure_ddpm_out
81
 
82
  editing_args = dict(
83
  editing_prompt = [edit_concept],