Linoy Tsaban commited on
Commit
ca0ffc2
1 Parent(s): e63f6d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -123,9 +123,8 @@ with gr.Blocks(css='style.css') as demo:
123
  zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
124
  xt = gr.State(value=wts[skip])
125
  zs = gr.State(value=zs[skip:])
126
- xt.value
127
 
128
- output = sample(zs, xt, prompt_tar=tar_prompt, cfg_scale_tar=cfg_scale_tar)
129
 
130
  return output, xt, zs
131
 
 
123
  zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=cfg_scale_src)
124
  xt = gr.State(value=wts[skip])
125
  zs = gr.State(value=zs[skip:])
 
126
 
127
+ output = sample(zs.value, xt.value, prompt_tar=tar_prompt, cfg_scale_tar=cfg_scale_tar)
128
 
129
  return output, xt, zs
130