Spaces:

xinjjj
/

ImgRoboAssetGen

Running on Zero

xinjie.wang commited on 21 days ago

Commit

755a578

1 Parent(s): 2674f7f

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -305,7 +305,7 @@ with gr.Blocks(
     )
     image_prompt.upload(
-        partial(preprocess_image_fn, model=gr.State(lambda: RBG_REMOVER), buffer=gr.State(lambda: IMAGE_BUFFER)),
         inputs=[image_prompt],
         outputs=[image_prompt],
     )

     )
     image_prompt.upload(
+        partial(preprocess_image_fn, model=gr.State(value=RBG_REMOVER), buffer=gr.State(value=IMAGE_BUFFER)),
         inputs=[image_prompt],
         outputs=[image_prompt],
     )