Update app.py

app.py

@@ -6,23 +6,42 @@ from diffusers import StableDiffusionXLPipeline
 from insightface.app import FaceAnalysis
 from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 import os
-import zipfile
+import urllib.request
 
-# Force offline mode
+# Force offline mode for Hugging Face Hub (but allow InsightFace download)
 os.environ["HF_HUB_OFFLINE"] = "1"
 
 # Set device to CPU
 device = "cpu"
 dtype = torch.float32
 
-# Extract InsightFace model if needed
+# Set up InsightFace model directory
 insightface_model_dir = "/home/user/.insightface/models/buffalo_l"
-if not os.path.exists(insightface_model_dir):
-    os.makedirs(insightface_model_dir, exist_ok=True)
-    with zipfile.ZipFile("./buffalo_l.zip", "r") as zip_ref:
+os.makedirs(insightface_model_dir, exist_ok=True)
+buffalo_l_zip = "./buffalo_l.zip"
+
+# Download buffalo_l.zip if not present
+if not os.path.exists(buffalo_l_zip):
+    try:
+        print("Downloading buffalo_l.zip for InsightFace...")
+        urllib.request.urlretrieve(
+            "https://github.com/deepinsight/insightface/releases/download/v0.7/buffalo_l.zip",
+            buffalo_l_zip
+        )
+        print("Download completed.")
+    except Exception as e:
+        print(f"Failed to download buffalo_l.zip: {e}")
+        # Fallback to a lighter model or raise error
+        raise RuntimeError("Cannot download buffalo_l.zip. Please ensure network access or preload the file.")
+
+# Extract buffalo_l.zip
+if os.path.exists(buffalo_l_zip):
+    import zipfile
+    with zipfile.ZipFile(buffalo_l_zip, "r") as zip_ref:
         zip_ref.extractall(insightface_model_dir)
+    print("Extracted buffalo_l.zip.")
 
-# Load face encoder with preloaded model
+# Load face encoder
 face_app = FaceAnalysis(providers=["CPUExecutionProvider"], root="/home/user/.insightface/models")
 face_app.prepare(ctx_id=0, det_size=(480, 480))