Update app.py

app.py

@@ -14,7 +14,7 @@ os.environ["HF_HUB_OFFLINE"] = "1"
 device = "cpu"
 dtype = torch.float32
 
-# Load face encoder (InsightFace will download buffalo_l if needed)
+# Load face encoder (InsightFace handles its own download)
 try:
     face_app = FaceAnalysis(providers=["CPUExecutionProvider"])
     face_app.prepare(ctx_id=0, det_size=(480, 480))
@@ -23,9 +23,13 @@ except Exception as e:
     raise RuntimeError(f"Failed to load InsightFace model: {e}. Ensure network access for initial download.")
 
 # Define paths for preloaded weights
-model_path = "./unet/"  # Adjusted for preloaded Kolors weights
+model_path = "./unet/"  # Preloaded Kolors weights should be at ./unet/
 ip_adapter_path = "./"
 
+# Debug: List files in the root directory to confirm preloading
+print("Files in root directory:", os.listdir("."))
+print("Files in ./unet/ directory:", os.listdir("./unet") if os.path.exists("./unet") else "No ./unet/ directory")
+
 # Check if weights exist
 kolors_weights = model_path + "diffusion_pytorch_model.safetensors"
 if not os.path.exists(kolors_weights):