Update model.py

model.py

@@ -11,7 +11,7 @@ from diffusers import (
 # Import the custom pipeline from your local file
 from pipeline_stable_diffusion_xl_instantid_img2img import StableDiffusionXLInstantIDImg2ImgPipeline
 
-from huggingface_hub import snapshot_download, hf_hub_download # <-- ADDED hf_hub_download
+from huggingface_hub import snapshot_download
 from insightface.app import FaceAnalysis
 from controlnet_aux import ZoeDetector, LineartDetector
 
@@ -65,28 +65,13 @@ class ModelHandler:
         # 2. Load ControlNets
         print("Loading ControlNets (InstantID, Zoe, LineArt)...")
 
-        # --- START FIX for OSError ---
-        # We must first download the InstantID ControlNet files to a local path.
-        controlnet_local_path = "./models/InstantID-ControlNet"
-        if not os.path.exists(os.path.join(controlnet_local_path, "config.json")):
-            print(f"Downloading InstantID ControlNet to {controlnet_local_path}...")
-            hf_hub_download(
-                repo_id=Config.INSTANTID_REPO,
-                filename="controlnet/config.json",
-                local_dir=controlnet_local_path,
-                local_dir_use_symlinks=False
-            )
-            hf_hub_download(
-                repo_id=Config.INSTANTID_REPO,
-                filename="controlnet/diffusion_pytorch_model.safetensors",
-                local_dir=controlnet_local_path,
-                local_dir_use_symlinks=False
-            )
-        
-        # Now, load the ControlNet from the LOCAL directory
-        print("Loading InstantID ControlNet from local files...")
+        # --- START FIX ---
+        # Load the InstantID ControlNet from the correct subfolder
+        # pointed out by the user.
+        print("Loading InstantID ControlNet from subfolder 'ControlNetModel'...")
         cn_instantid = ControlNetModel.from_pretrained(
-            controlnet_local_path, # <-- Load from local path
+            Config.INSTANTID_REPO,   # "InstantX/InstantID"
+            subfolder="ControlNetModel", # <-- THE FIX
             torch_dtype=Config.DTYPE
         )
         print("  [OK] Loaded InstantID ControlNet.")
@@ -141,7 +126,7 @@ class ModelHandler:
                 return None
                 
             # Sort by size (width * height) to find the main character
-            faces = sorted(faces, key=lambda x: (x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x['bbox'][1]), reverse=True)
+            faces = sorted(faces, key=lambda x: (x['bbox'][2]-x['bbox'][0])*(x['bbox'][3]-x_bbox[1]), reverse=True)
             
             # Return the largest face
             return torch.tensor(faces[0].normed_embedding).unsqueeze(0)