Update app.py

app.py

@@ -253,7 +253,7 @@ def build_model_lora(pipe_concept, lora_paths, style_path, condition, args, pipe
 def build_yolo_segment_model(sam_path, device):
     yolo_world = YOLOWorld(model_id="yolo_world/l")
     sam = EfficientViTSamPredictor(
-        create_sam_model(name='efficientvit_sam_xl1.pt', weight_url=sam_path).to(device).eval()
+        create_sam_model(name='xl1', weight_url=sam_path).to(device).eval()
     )
     return yolo_world, sam
 
@@ -577,7 +577,7 @@ def parse_args():
     parser.add_argument('--openpose_checkpoint', default='thibaud/controlnet-openpose-sdxl-1.0', type=str)
     parser.add_argument('--canny_checkpoint', default='diffusers/controlnet-canny-sdxl-1.0', type=str)
     parser.add_argument('--depth_checkpoint', default='diffusers/controlnet-depth-sdxl-1.0', type=str)
-    parser.add_argument('--efficientViT_checkpoint', default='./checkpoint/sam/xl1.pt', type=str)
+    parser.add_argument('--efficientViT_checkpoint', default='./checkpoint/sam/efficientvit_sam_xl1.pt', type=str)
     parser.add_argument('--dino_checkpoint', default='./checkpoint/GroundingDINO', type=str)
     parser.add_argument('--sam_checkpoint', default='./checkpoint/sam/sam_vit_h_4b8939.pth', type=str)
     parser.add_argument('--dpt_checkpoint', default='Intel/dpt-hybrid-midas', type=str)