Update app.py

app.py

@@ -12,26 +12,33 @@ class ObjectDetector:
         self.api_url = "https://api-inference.huggingface.co/models/facebook/detr-resnet-50"
         
     def detect(self, image, hf_token=None):
+        import base64
+    
         # Try multiple ways to get HF token
         token = hf_token or os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
-        
         if not token:
             raise Exception("HF Token required. Please set HF_TOKEN in Space secrets or environment variables")
-            
-        headers = {"Authorization": f"Bearer {token}"}
-        
-        # Convert PIL image to bytes
+    
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "Content-Type": "application/json"
+        }
+    
+        # Convert PIL image to base64 string
         img_buffer = io.BytesIO()
         image.save(img_buffer, format='JPEG')
         img_bytes = img_buffer.getvalue()
-        
-        response = requests.post(self.api_url, headers=headers, data=img_bytes)
-        
+        img_base64 = base64.b64encode(img_bytes).decode("utf-8")
+    
+        # Send JSON payload with base64 image
+        response = requests.post(self.api_url, headers=headers, json={"inputs": img_base64})
+    
         if response.status_code != 200:
             raise Exception(f"Object detection API error: {response.status_code} - {response.text}")
-            
+    
         return response.json()
 
+
 object_detector = ObjectDetector()
 
 # COCO class names for object detection