Update app.py

app.py

@@ -50,25 +50,18 @@ else:
 
 logging.info(f"Running on device={device} with dtype={torch_dtype}")
 
-# --- Model IDs & Token ---
-huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
-if not huggingface_token:
-    print("ERROR: Environment variable HUGGINGFACE_TOKEN not set. Please add your token in Settings → Secrets.")
-    sys.exit(1)
-
+# --- Model IDs & Download (no token) ---
 flux_model_id       = "black-forest-labs/FLUX.1-dev"
 controlnet_model_id = "jasperai/Flux.1-dev-Controlnet-Upscaler"
 local_model_dir     = flux_model_id.split("/")[-1]
 pipe = None
 
-# --- Download & Load Models ---
 try:
-    logging.info(f"Downloading {flux_model_id} with token…")
+    logging.info(f"Downloading base model: {flux_model_id}")
     model_path = snapshot_download(
         repo_id=flux_model_id,
         repo_type="model",
         local_dir=local_model_dir,
-        use_auth_token=huggingface_token,
         ignore_patterns=["*.md", "*.gitattributes"],
     )
     logging.info(f"Downloaded base model to: {model_path}")
@@ -76,8 +69,7 @@ try:
     logging.info(f"Loading ControlNet: {controlnet_model_id}")
     controlnet = FluxControlNetModel.from_pretrained(
         controlnet_model_id,
-        torch_dtype=torch_dtype,
-        use_auth_token=huggingface_token
+        torch_dtype=torch_dtype
     ).to(device)
     logging.info("ControlNet loaded.")
 
@@ -85,18 +77,13 @@ try:
     pipe = FluxControlNetPipeline.from_pretrained(
         model_path,
         controlnet=controlnet,
-        torch_dtype=torch_dtype,
-        use_auth_token=huggingface_token
+        torch_dtype=torch_dtype
     ).to(device)
     logging.info("Pipeline ready.")
 
 except Exception as e:
     logging.error(f"Error loading models: {e}", exc_info=True)
-    # Detect gated‐repo 401 by checking for “401” in the message
-    if "401" in str(e):
-        print(f"FATAL: Cannot access gated model {flux_model_id}. Have you accepted its license and set a valid HUGGINGFACE_TOKEN?")
-    else:
-        print(f"FATAL: model load failed: {e}")
+    print(f"FATAL: could not load models: {e}")
     sys.exit(1)
 
 # --- Constants & Helpers ---