Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,8 +9,11 @@ import torch
 from transformers import DetrImageProcessor, DetrForObjectDetection
 from huggingface_hub import login
 
-# Get token from environment (set as secret in HF Spaces)
+# Secure token handling
 HUGGINGFACE_TOKEN = os.environ.get("HF_TOKEN")
+if HUGGINGFACE_TOKEN is None:
+    raise ValueError("Hugging Face token not found in environment. Please set HF_TOKEN in Space secrets.")
+
 login(token=HUGGINGFACE_TOKEN)
 
 # === Load Model and Processor ===