Update app.py

app.py

@@ -91,16 +91,14 @@ def initialize_models(sam_path, adapter_pth, model_cfg, cfg):
     # 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")
         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)
-        adapter_state_dict = {k.replace('mask_adapter.', '').replace('adapter.', ''): v 
-                              for k, v in adapter_state_dict["model"].items() 
-                              if k.startswith('adapter') or k.startswith('mask_adapter')}
+        adapter_state_dict = torch.load(adapter_pth, map_location=torch.device('cpu'))
         mask_adapter.load_state_dict(adapter_state_dict)
         print("Mask Adapter model initialized.")