Spaces:

CUHKWilliam
/

Multiple-shots

Sleeping

CUHKWilliam commited on Jun 4, 2024

Commit

56cf879

verified ·

1 Parent(s): 7ec5dd7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -92,7 +92,7 @@ def inference_mask1(
 inputs = [gr.Image(label='query')]
 for i in range(5):
-    inputs.append(gr.ImageMask(label='prompt{}'.format(i)))
 demo_mask = gr.Interface(fn=inference_mask1,
                    inputs=inputs,
     outputs=[gr.Image(label="output")],

 inputs = [gr.Image(label='query')]
 for i in range(5):
+    inputs.append(gr.ImageMask(label='support {}'.format(i)))
 demo_mask = gr.Interface(fn=inference_mask1,
                    inputs=inputs,
     outputs=[gr.Image(label="output")],