download model

handler.py

@@ -35,6 +35,12 @@ class EndpointHandler():
         print("Loading model from", model_dir)
         
         hf_hub_download(repo_id="InstantX/InstantID", filename="ControlNetModel/config.json", local_dir="./checkpoints")
+        hf_hub_download(
+            repo_id="InstantX/InstantID",
+            filename="ControlNetModel/diffusion_pytorch_model.safetensors",
+            local_dir="./checkpoints",
+        )
+        hf_hub_download(repo_id="InstantX/InstantID", filename="ip-adapter.bin", local_dir="./checkpoints")
 
         face_adapter = f"./checkpoints/ip-adapter.bin"
         controlnet_path = f"./checkpoints/ControlNetModel"

requirements.txt

@@ -14,4 +14,5 @@ insightface
 gradio
 controlnet_aux
 gdown
-peft
+peft
+mediapipe