Upload web-ui.py

web-ui.py

@@ -13,7 +13,7 @@ from insightface.utils import face_align
 from pyngrok import ngrok
 import threading
 import time
-from ip_adapter.ip_adapter_faceid import IPAdapterFaceIDXL
+from ip_adapter.ip_adapter_faceid import IPAdapterFaceIDPlusXL
 import hashlib
 from datetime import datetime
 
@@ -116,8 +116,9 @@ def load_model(model_name, isSDXL):
         ).to(device)
 
     if isSDXL:
-        ip_ckpt = "adapters/ip-adapter-faceid_sdxl.bin"
-        ip_model = IPAdapterFaceIDXL(pipe, ip_ckpt, device)
+        image_encoder_path = "h94/IP-Adapter/models/image_encoder"
+        ip_ckpt = "adapters/ip-adapter-faceid-plusv2_sdxl.bin"
+        ip_model = IPAdapterFaceIDPlusXL(pipe,image_encoder_path, ip_ckpt, device)
     else:
         image_encoder_path = "h94/IP-Adapter/models/image_encoder"
         ip_ckpt = "adapters/ip-adapter-faceid-plusv2_sd15.bin"