Update app.py

app.py

@@ -29,8 +29,10 @@ from utils.controlnet_xs import ControlNetXSAdapter
 
 import gradio as gr
 
-# import torch.nn.functional as F
-# from torchvision.transforms import Compose
+hf_hub_download(repo_id="RED-AIGC/InstantID-XS", filename="controlnetxs.bin", local_dir="./ckpt")
+hf_hub_download(repo_id="RED-AIGC/InstantID-XS",filename="cross_attn.bin",local_dir="./ckpt",)
+hf_hub_download(repo_id="RED-AIGC/InstantID-XS", filename="image_proj.bin", local_dir="./ckpt")
+
 
 # global variable
 MAX_SEED = np.iinfo(np.int32).max
@@ -42,9 +44,9 @@ base_model = 'frankjoshua/realvisxlV40_v40Bakedvae'
 vae_path = 'madebyollin/sdxl-vae-fp16-fix'
 ckpt = 'RED-AIGC/InstantID-XS'
 
-image_proj_path = os.path.join(ckpt, "image_proj.bin")
-cnxs_path =  os.path.join(ckpt, "controlnetxs.bin")
-cross_attn_path = os.path.join(ckpt, "cross_attn.bin")
+image_proj_path = "./ckpt/image_proj.bin")
+cnxs_path =  "./ckpt/controlnetxs.bin")
+cross_attn_path = "./ckpt/cross_attn.bin")
 
 
 # Load face encoder