Update app.py

app.py

@@ -96,7 +96,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, map_location=torch.device('cpu'))
         mask_adapter.load_state_dict(adapter_state_dict)