xinjie.wang commited on
Commit
2674f7f
·
1 Parent(s): 55ed985
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -305,7 +305,7 @@ with gr.Blocks(
305
  )
306
 
307
  image_prompt.upload(
308
- partial(preprocess_image_fn, model=RBG_REMOVER, buffer=IMAGE_BUFFER),
309
  inputs=[image_prompt],
310
  outputs=[image_prompt],
311
  )
 
305
  )
306
 
307
  image_prompt.upload(
308
+ partial(preprocess_image_fn, model=gr.State(lambda: RBG_REMOVER), buffer=gr.State(lambda: IMAGE_BUFFER)),
309
  inputs=[image_prompt],
310
  outputs=[image_prompt],
311
  )