Linoy Tsaban commited on
Commit
fb716a5
1 Parent(s): 2ff138d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -121,6 +121,9 @@ with gr.Blocks(css='style.css') as demo:
121
  if not wt:
122
  # invert and retrieve noise maps and latent
123
  wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
 
 
 
124
  output = sample(wt, zs, wts, prompt_tar=src_prompt, cfg_scale_tar=cfg_scale_tar, skip=skip)
125
  return output
126
 
 
121
  if not wt:
122
  # invert and retrieve noise maps and latent
123
  wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
124
+ wt = gr.State(value=wt)
125
+ zs = gr.State(value=zs)
126
+ wts = gr.State(value=wts)
127
  output = sample(wt, zs, wts, prompt_tar=src_prompt, cfg_scale_tar=cfg_scale_tar, skip=skip)
128
  return output
129