Upload app.py

app.py

@@ -179,7 +179,6 @@ def segment_and_overlay(image: Image.Image, points):
     overlaid = apply_mask_overlay(image, mask)
     return overlaid
 
-
 def reset_points():
     """
     清空 prompt 点
@@ -355,9 +354,9 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             points_state = gr.State([])
             points_output = gr.Textbox(label="Target Object Prompts", interactive=False)
             # 为 image_annotation 添加点击事件记录 prompt 点
-            image_annotation.click(
+            image_annotation.select(
                 record_click,
-                inputs=[gr.EventData(), points_state],
+                inputs=[points_state],
                 outputs=[points_state, points_output]
             )
             # 新增：分割后展示结果的组件
@@ -489,7 +488,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    sam_checkpoint = hf_hub_download(repo_id="ybelkada/segment-anything", filename="sam_vit_h_4b8939.pth")
+    sam_checkpoint = hf_hub_download("ybelkada/segment-anything", "checkpoints/sam_vit_h_4b8939.pth")
     model_type = "vit_h"
     sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
     sam.cuda()