Update app.py

app.py

@@ -14,23 +14,13 @@ os.environ["HF_HUB_OFFLINE"] = "1"
 device = "cpu"
 dtype = torch.float32
 
-# Set up InsightFace model directory
-insightface_model_dir = "/home/user/.insightface/models/buffalo_l"
-os.makedirs(insightface_model_dir, exist_ok=True)
-buffalo_l_zip = "./buffalo_l.zip"
-
-# Check and extract buffalo_l.zip (runtime download removed due to network issues)
-if not os.path.exists(buffalo_l_zip):
-    raise FileNotFoundError(f"buffalo_l.zip not found at {buffalo_l_zip}. Preloading failed. Please check .env configuration.")
-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
-face_app = FaceAnalysis(providers=["CPUExecutionProvider"], root="/home/user/.insightface/models")
-face_app.prepare(ctx_id=0, det_size=(480, 480))
+# Load face encoder (InsightFace will download buffalo_l if needed)
+try:
+    face_app = FaceAnalysis(providers=["CPUExecutionProvider"])
+    face_app.prepare(ctx_id=0, det_size=(480, 480))
+    print("InsightFace model loaded successfully.")
+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