Update app.py

app.py

@@ -12,11 +12,13 @@ UPLOAD_FOLDER = 'static/captures'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
-# Hugging Face API settings (replace with your API key)
-HF_API_TOKEN = "your_hugging_face_api_token"  # Get from https://huggingface.co/settings/tokens
+# Hugging Face API settings
+HF_API_TOKEN = os.getenv('HF_API_TOKEN')  # Load token from environment variable
 HF_API_URL = "https://api-inference.huggingface.co/models/facebook/detr-resnet-50"  # Example model for object detection
 
 def query_hugging_face(image_data):
+    if not HF_API_TOKEN:
+        raise ValueError("Hugging Face API token not set. Please set the HF_API_TOKEN environment variable.")
     headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
     response = requests.post(HF_API_URL, headers=headers, data=image_data)
     return response.json()