radames commited on
Commit
8de3410
1 Parent(s): 5486fe5
Files changed (2) hide show
  1. .gitignore +2 -1
  2. app.py +1 -1
.gitignore CHANGED
@@ -1,2 +1,3 @@
1
  __pycache__/
2
- *.log
 
 
1
  __pycache__/
2
+ *.log
3
+ gradio_cached_examples
app.py CHANGED
@@ -143,7 +143,7 @@ def predict(
143
  )
144
 
145
  rgb_img = tensor_to_pil(vaedecode_sample[0])
146
- return (rgb_img[0], None, seed)
147
 
148
  else:
149
  layereddiffusionapply_sample = ld_fg_apply_layered_diffusion(
 
143
  )
144
 
145
  rgb_img = tensor_to_pil(vaedecode_sample[0])
146
+ return (rgb_img[0], rgb_img[0], seed)
147
 
148
  else:
149
  layereddiffusionapply_sample = ld_fg_apply_layered_diffusion(