Update app.py

app.py

@@ -85,20 +85,20 @@ 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="cpu", apply_postprocessing=False)
+        sam2_model = build_sam2(model_cfg, sam_path, device="cuda", apply_postprocessing=False)
         print("SAM2 model initialized.")
     
     # CLIP模型初始化
     if clip_model is None:
         clip_model, _, _ = open_clip.create_model_and_transforms("convnext_large_d_320", pretrained="laion2b_s29b_b131k_ft_soup")
-        clip_model = clip_model.to("cpu")
+        clip_model = clip_model.to("cuda")
         print("CLIP model initialized.")
     
     # Mask Adapter模型初始化
     if mask_adapter is None:
-        mask_adapter = build_mask_adapter(cfg, "MASKAdapterHead").cpu()
+        mask_adapter = build_mask_adapter(cfg, "MASKAdapterHead").cuda()
         # 加载Adapter状态字典
-        adapter_state_dict = torch.load(adapter_pth, map_location=torch.device('cpu'))
+        adapter_state_dict = torch.load(adapter_pth)
         mask_adapter.load_state_dict(adapter_state_dict)
         print("Mask Adapter model initialized.")