Update app.py

app.py

@@ -16,14 +16,12 @@ import sys
 
 
 def add_sam2_to_path():
-    """将SAM2安装目录添加到Python路径"""
     sam2_dir = os.path.abspath("third_party/sam2")
     if sam2_dir not in sys.path:
         sys.path.insert(0, sam2_dir)
     return sam2_dir
 
 def install_sam2():
-    """检查并安装SAM2及其依赖"""
     sam2_dir = "third_party/sam2"
     if not os.path.exists(sam2_dir):
         print("Installing SAM2...")
@@ -85,12 +83,8 @@ except Exception as e:
 # SAM Wrapper
 class CustomSAMWrapper:
     def __init__(self, model_path: str, device: str = DEVICE):
-        try:
-            from sam2.build_sam import build_sam2
-            from sam2.sam2_image_predictor import SAM2ImagePredictor
-            
+        try:            
             self.device = torch.device(device)
-            # model_cfg = os.path.join("third_party/sam2", "configs/sam2.1/sam2.1_hiera_l.yaml")
             sam_model = build_sam2("configs/sam2.1/sam2.1_hiera_l.yaml", model_path)
             sam_model = sam_model.to(self.device)
             self.predictor = SAM2ImagePredictor(sam_model)
@@ -317,8 +311,9 @@ def run_pipeline(image: PILImage.Image, prompt: str):
 
 
 with gr.Blocks(title="Seg-R1") as demo:
-    gr.Markdown("# Seg-R1")
+    # gr.Markdown("# Seg-R1")
     # gr.Markdown("Upload an image and ask questions about segmentation.")
+    gr.HTML(load_description("assets/title.md"))
     
     with gr.Row():
         with gr.Column():