Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import CLIPVisionModelWithProjection,CLIPImageProcessor
 from diffusers.utils import load_image
 import os,sys
 import gradio as gr
+from huggingface_hub import hf_hub_download
 
 from kolors.pipelines.pipeline_controlnet_xl_kolors_img2img_face import StableDiffusionXLControlNetImg2ImgPipeline
 from kolors.models.modeling_chatglm import ChatGLMModel
@@ -64,10 +65,14 @@ controlnet = ControlNetModel.from_pretrained( control_path , revision=None).to(d
 
 face_info_generator = FaceInfoGenerator(root_dir = "./")
 
-clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",cache_dir='./',subfolder="clip-vit-large-patch14-336", ignore_mismatched_sizes=True)
+clip_image_encoder = CLIPVisionModelWithProjection.from_pretrained("Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",subfolder="clip-vit-large-patch14-336", ignore_mismatched_sizes=True)
 clip_image_processor = CLIPImageProcessor(size = 336, crop_size = 336)
+
+hf_hub_download(repo_id="Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus", filename="ipa-faceid-plus.bin",cache_dir='./')
+
 snapshotname = os.listdir('./models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots')[0]
 print(os.listdir(f'models--Kwai-Kolors--Kolors-IP-Adapter-FaceID-Plus/snapshots/{snapshotname}'))
+
 pipe = StableDiffusionXLControlNetImg2ImgPipeline(
         vae=vae,
         controlnet = controlnet,