add sam2 tab

app.py

@@ -120,6 +120,10 @@ def rmbg(image=None, url=None):
     return image
 
 
+def mask_generation(image=None, json=None):
+    return None
+
+
 @spaces.GPU
 def main(*args):
     api_num = args[0]
@@ -130,12 +134,14 @@ def main(*args):
         return outpaint(*args)
     elif api_num == 3:
         return inpaint(*args)
+    elif api_num == 4:
+        return mask_generation(*args)
 
 
 rmbg_tab = gr.Interface(
     fn=main,
     inputs=[
-        gr.Number(1, visible=False),
+        gr.Number(1, interactive=False),
         "image",
         gr.Text("", label="url"),
     ],
@@ -149,7 +155,7 @@ rmbg_tab = gr.Interface(
 outpaint_tab = gr.Interface(
     fn=main,
     inputs=[
-        gr.Number(2, visible=False),
+        gr.Number(2, interactive=False),
         gr.Image(label="image", type="pil"),
         gr.Number(label="padding top"),
         gr.Number(label="padding bottom"),
@@ -169,7 +175,7 @@ outpaint_tab = gr.Interface(
 inpaint_tab = gr.Interface(
     fn=main,
     inputs=[
-        gr.Number(3, visible=False),
+        gr.Number(3, interactive=False),
         gr.Image(label="image", type="pil"),
         gr.Image(label="mask", type="pil"),
         gr.Text(label="prompt"),
@@ -183,9 +189,20 @@ inpaint_tab = gr.Interface(
     description="it is recommended that you use https://github.com/la-voliere/react-mask-editor when creating an image mask in JS and then inverse it before sending it to this space",
 )
 
+
+sam2_tab = gr.Interface(
+    main,
+    inputs=[
+        gr.Number(4, interactive=False),
+        gr.Image("image", type="pil"),
+        gr.JSON(),
+    ],
+    outputs=["image"],
+)
+
 demo = gr.TabbedInterface(
-    [rmbg_tab, outpaint_tab, inpaint_tab],
-    ["remove background", "outpainting", "inpainting"],
+    [rmbg_tab, outpaint_tab, inpaint_tab, sam2_tab],
+    ["remove background", "outpainting", "inpainting", "sam2"],
     title="Utilities that require GPU",
 )