Update app.py

app.py

@@ -6,9 +6,8 @@ from diffusers import StableDiffusionXLPipeline
 from insightface.app import FaceAnalysis
 from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 import os
-import urllib.request
 
-# Force offline mode for Hugging Face Hub (but allow InsightFace download)
+# Force offline mode for Hugging Face Hub
 os.environ["HF_HUB_OFFLINE"] = "1"
 
 # Set device to CPU
@@ -20,20 +19,9 @@ insightface_model_dir = "/home/user/.insightface/models/buffalo_l"
 os.makedirs(insightface_model_dir, exist_ok=True)
 buffalo_l_zip = "./buffalo_l.zip"
 
-# Download buffalo_l.zip if not present
+# Check and extract buffalo_l.zip (runtime download removed due to network issues)
 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}")
-        raise RuntimeError("Cannot download buffalo_l.zip. Please ensure network access.")
-
-# Extract 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:
@@ -48,23 +36,10 @@ face_app.prepare(ctx_id=0, det_size=(480, 480))
 model_path = "./unet/"  # Adjusted for preloaded Kolors weights
 ip_adapter_path = "./"
 
-# Check if Kolors weights exist, download if missing
+# Check if weights exist
 kolors_weights = model_path + "diffusion_pytorch_model.safetensors"
-if not os.path.exists(model_path):
-    os.makedirs(model_path, exist_ok=True)
 if not os.path.exists(kolors_weights):
-    try:
-        print("Downloading Kolors model weights...")
-        urllib.request.urlretrieve(
-            "https://huggingface.co/Kwai-Kolors/Kolors-diffusers/resolve/main/unet/diffusion_pytorch_model.safetensors",
-            kolors_weights
-        )
-        print("Kolors weights downloaded.")
-    except Exception as e:
-        print(f"Failed to download Kolors weights: {e}")
-        raise RuntimeError("Cannot download Kolors weights. Please ensure network access.")
-
-# Check if IP-Adapter weights exist
+    raise FileNotFoundError(f"Kolors model weights not found at {kolors_weights}. Preloading failed. Please check .env configuration.")
 if not os.path.exists(ip_adapter_path + "ip-adapter.bin"):
     raise FileNotFoundError(f"IP-Adapter weights not found at {ip_adapter_path}")
 
@@ -97,7 +72,7 @@ def generate_image(uploaded_image, prompt):
             guidance_scale=7.5,
             height=512,
             width=512,
-        ).images[0]  # Closing parenthesis here
+        ).images[0]
         return "Image generated successfully!", image
     except Exception as e:
         return f"Generation failed: {e}", None