Upload app.py

app.py

@@ -140,7 +140,6 @@ def button_clickable(selected_points):
         return gr.Button.update(interactive=False)
 
 
-@spaces.GPU(duration=90)
 def run_sam(predictor: SamPredictor, selected_points):
     """
     调用 SAM 模型进行分割。
@@ -158,7 +157,6 @@ def run_sam(predictor: SamPredictor, selected_points):
     visible_mask = 255 * np.squeeze(masks).astype(np.uint8)
     return visible_mask, None
 
-@spaces.GPU
 def apply_mask_overlay(image, mask):
     """
     在原图上叠加 mask：使用红色绘制 mask 的轮廓，非 mask 区域叠加浅灰色半透明遮罩。
@@ -174,7 +172,6 @@ def apply_mask_overlay(image, mask):
     cv2.drawContours(overlay, contours, -1, (255, 0, 0), 2)
     return overlay
 
-@spaces.GPU
 def segment_and_overlay(image, points, sam_predictor):
     """
     调用 run_sam 获得 mask，然后叠加显示分割结果。
@@ -349,7 +346,6 @@ def split_image(image: Image.Image) -> list:
         images.append(Image.fromarray(image[:, s:e+1]))
     return [image for image in images]
 
-@spaces.GPU
 def get_sam_predictor():
     sam_checkpoint = hf_hub_download("ybelkada/segment-anything", "checkpoints/sam_vit_h_4b8939.pth")
     model_type = "vit_h"