add spaces

app.py

@@ -16,6 +16,7 @@ from matplotlib import cm
 from glob import glob
 from natsort import natsorted
 from huggingface_hub import HfApi, upload_file
+import spaces
 
 # ===== 导入三个推理模块 =====
 from inference_seg import load_model as load_seg_model, run as run_seg
@@ -210,6 +211,7 @@ def colorize_mask(mask: np.ndarray, num_colors: int = 512) -> np.ndarray:
     return palette_arr[color_idx]
 
 # ===== 分割功能 =====
+@spaces.GPU
 def segment_with_choice(use_box_choice, annot_value):
     """分割主函数 - 每个实例不同颜色+轮廓"""
     if annot_value is None or len(annot_value) < 1:
@@ -301,6 +303,7 @@ def segment_with_choice(use_box_choice, annot_value):
     return Image.fromarray(overlay), temp_mask_file.name
 
 # ===== 计数功能 =====
+@spaces.GPU
 def count_cells_handler(use_box_choice, annot_value):
     """Counting handler - supports bounding box, returns only density map"""
     if annot_value is None or len(annot_value) < 1:
@@ -704,7 +707,7 @@ def create_tracking_visualization(tif_dir, output_dir, valid_tif_files):
         except:
             return valid_tif_files[0]
 
-
+@spaces.GPU
 def track_video_handler(use_box_choice, first_frame_annot, zip_file_obj):
     """
     支持 ZIP 压缩包上传的 Tracking 处理函数 - 支持首帧边界框