Update app.py

app.py

@@ -660,19 +660,19 @@ def main(device, segment_type):
 
 def parse_args():
     parser = argparse.ArgumentParser('', add_help=False)
-    parser.add_argument('--pretrained_model', default='/home/data1/kz_dir/checkpoint/YamerMIX_v8', type=str)
-    parser.add_argument('--controlnet_path', default='../checkpoint/InstantID/ControlNetModel', type=str)
-    parser.add_argument('--face_adapter_path', default='../checkpoint/InstantID/ip-adapter.bin', type=str)
-    parser.add_argument('--openpose_checkpoint', default='../checkpoint/controlnet-openpose-sdxl-1.0', type=str)
-    parser.add_argument('--canny_checkpoint', default='../checkpoint/controlnet-canny-sdxl-1.0', type=str)
-    parser.add_argument('--depth_checkpoint', default='../checkpoint/controlnet-depth-sdxl-1.0', type=str)
-    parser.add_argument('--dpt_checkpoint', default='../checkpoint/dpt-hybrid-midas', type=str)
+    parser.add_argument('--pretrained_model', default='wangqixun/YamerMIX_v8', type=str)
+    parser.add_argument('--controlnet_path', default='./checkpoint/InstantID/ControlNetModel', type=str)
+    parser.add_argument('--face_adapter_path', default='./checkpoint/InstantID/ip-adapter.bin', type=str)
+    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('--dpt_checkpoint', default='Intel/dpt-hybrid-midas', type=str)
     parser.add_argument('--pose_detector_checkpoint',
-                        default='../checkpoint/ControlNet/annotator/ckpts/body_pose_model.pth', type=str)
-    parser.add_argument('--efficientViT_checkpoint', default='../checkpoint/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('--antelopev2_path', default='../checkpoint/antelopev2', type=str)
+                        default='./checkpoint/ControlNet/annotator/ckpts/body_pose_model.pth', type=str)
+    parser.add_argument('--efficientViT_checkpoint', default='./checkpoint/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('--antelopev2_path', default='./checkpoint/antelopev2', type=str)
     parser.add_argument('--save_dir', default='results/instantID', type=str)
     parser.add_argument('--prompt', default='Close-up photo of the cool man and beautiful woman as they accidentally discover a mysterious island while on vacation by the sea, facing the camera smiling, 35mm photograph, film, professional, 4k, highly detailed.', type=str)
     parser.add_argument('--negative_prompt', default='noisy, blurry, soft, deformed, ugly', type=str)