Auto-login with HF_TOKEN environment variable for Spaces

app.py

@@ -13,6 +13,15 @@ import torch
 from huggingface_hub import hf_hub_download, scan_cache_dir, login, whoami
 from safetensors import safe_open
 
+# Auto-login with HF_TOKEN if available (for HuggingFace Spaces)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if HF_TOKEN:
+    try:
+        login(token=HF_TOKEN, add_to_git_credential=False)
+        print("✅ Automatically logged in using HF_TOKEN")
+    except Exception as e:
+        print(f"⚠️ Auto-login failed: {str(e)}")
+
 
 def hf_login(token: str):
     """Login to Hugging Face with provided token."""
@@ -382,7 +391,11 @@ with gr.Blocks(title="Hugging Face Model Weight Inspector") as demo:
     gr.Markdown("# 🔍 Hugging Face Model Weight Inspector")
     
     # HF Login section
-    with gr.Accordion("🔐 Hugging Face Login (for private/gated models)", open=False):
+    with gr.Accordion("🔐 Hugging Face Login (Optional - auto-login with HF_TOKEN)", open=False):
+        gr.Markdown("""
+        **Note:** This Space automatically uses the `HF_TOKEN` secret if configured.  
+        Manual login below is only needed if you want to use a different token.
+        """)
         with gr.Row():
             with gr.Column(scale=3):
                 hf_token_input = gr.Textbox(