Update app.py

app.py

@@ -18,10 +18,14 @@ HF_API_URL = "https://api-inference.huggingface.co/models/facebook/detr-resnet-5
 
 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()
+        return {"error": "Hugging Face API token not set. Please set the HF_API_TOKEN environment variable."}
+    try:
+        headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
+        response = requests.post(HF_API_URL, headers=headers, data=image_data, timeout=10)
+        response.raise_for_status()  # Raise an error for bad status codes
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        return {"error": f"Failed to connect to Hugging Face API: {str(e)}"}
 
 @app.route('/')
 def index():