Update app.py

app.py

@@ -85,7 +85,7 @@ def initialize_models(sam_path, adapter_pth, model_cfg, cfg):
 
     # SAM2初始化
     if sam2_model is None:
-        sam2_model = build_sam2(model_cfg, sam_path, device="cuda", apply_postprocessing=False)
+        sam2_model = build_sam2(model_cfg, sam_path, device="cpu", apply_postprocessing=False)
         print("SAM2 model initialized.")
     
     # CLIP模型初始化
@@ -95,7 +95,7 @@ def initialize_models(sam_path, adapter_pth, model_cfg, cfg):
     
     # Mask Adapter模型初始化
     if mask_adapter is None:
-        mask_adapter = build_mask_adapter(cfg, "MASKAdapterHead").cuda()
+        mask_adapter = build_mask_adapter(cfg, "MASKAdapterHead").cpu()
         # 加载Adapter状态字典
         adapter_state_dict = torch.load(adapter_pth)
         adapter_state_dict = {k.replace('mask_adapter.', '').replace('adapter.', ''): v