Update app.py

app.py

@@ -14,6 +14,7 @@ import numpy as np
 # from sam2_mask import create_sam2_tab
 import subprocess
 import os
+import sam2_mask
 
 subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 HF_TOKEN = os.environ["HF_TOKEN"]
@@ -590,9 +591,9 @@ with gr.Blocks(css=css, fill_height=True) as demo:
         #             outputs=[image_output],
         #             show_progress=True,
         #         )
-        with gr.Tab("SAM2.1 Segmented Mask"):
-            with gr.Blocks():
-                masking_tool = sam_block
+        with gr.TabItem("SAM2.1 Segmented Mask"):
+            with gr.Blocks() as sam_demo:
+                sam2_mask.demo.render()
             # Define the action triggered by the upload_image_input change
             # upload_image_input.change(
             #     fn=update_mask,