Update app.py

app.py

@@ -35,7 +35,7 @@ def setup_cfg(config_file):
 
 def inference_automatic(input_img, class_names):
     mp.set_start_method("spawn", force=True)
-    config_file = '/home/yongkangli/Mask-Adapter/configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
+    config_file = './configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
     cfg = setup_cfg(config_file)
     
     demo = SAMVisualizationDemo(cfg, 0.8, sam2_model, clip_model,mask_adapter)
@@ -51,12 +51,12 @@ def inference_point(input_img, evt: gr.SelectData,):
     # In point mode, implement the logic to process points from the user click (x, y)
     # You can adjust your segmentation logic based on clicked points.
     x, y = evt.index[0], evt.index[1]
-    points = [[x, y]]  # 假设只选择一个点作为输入
+    points = [[x, y]]
     print(f"Selected point: {points}")
     import time
     start_time = time.time()
     mp.set_start_method("spawn", force=True)
-    config_file = '/home/yongkangli/Mask-Adapter/configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
+    config_file = './configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
     cfg = setup_cfg(config_file)
     
     demo = SAMPointVisualizationDemo(cfg, 0.8, sam2_model, clip_model,mask_adapter)
@@ -106,9 +106,9 @@ def initialize_models(sam_path, adapter_pth, model_cfg, cfg):
 
 # 初始化配置和模型
 model_cfg = "configs/sam2.1/sam2.1_hiera_l.yaml"
-sam_path = '/home/yongkangli/segment-anything-2/checkpoints/sam2.1_hiera_large.pt'
+sam_path = './sam2.1_hiera_large.pt'
 adapter_pth = './model_0279999_with_sem_new.pth'
-cfg = '/home/yongkangli/Mask-Adapter/configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
+cfg = './configs/ground-truth-warmup/mask-adapter/mask_adapter_convnext_large_cocopan_eval_ade20k.yaml'
 
 # 调用初始化函数
 initialize_models(sam_path, adapter_pth, model_cfg, cfg)