Spaces:

theSure
/

Omnieraser

Running on Zero

theSure commited on 17 days ago

Commit

ce3e13a

verified ·

1 Parent(s): 69dbad3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ def set_seed(seed):
 @spaces.GPU
 def predict(
     input_image,
-    uploaded_mask
     prompt,
     ddim_steps,
     seed,
@@ -170,7 +170,7 @@ def infer(
 ):
     return predict(input_image,
-                   uploaded_mask
                    removal_prompt,
                    ddim_steps,
                    seed,
@@ -325,7 +325,7 @@ with gr.Blocks(
         fn=infer,
         inputs=[
             input_image,
-            uploaded_mask
             ddim_steps,
             seed,
             scale,

 @spaces.GPU
 def predict(
     input_image,
+    uploaded_mask,
     prompt,
     ddim_steps,
     seed,
 ):
     return predict(input_image,
+                   uploaded_mask,
                    removal_prompt,
                    ddim_steps,
                    seed,
         fn=infer,
         inputs=[
             input_image,
+            uploaded_mask,
             ddim_steps,
             seed,
             scale,