Update app.py

app.py

@@ -22,14 +22,21 @@ MODEL_URL = "https://api-inference.huggingface.co/models/Salesforce/blip-image-c
 
 def get_caption(image_path):
     """Sends image to Hugging Face API for captioning"""
-    headers = {"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"}
+    headers = {
+        "Authorization": f"Bearer {HUGGINGFACE_API_KEY}",
+        "Content-Type": "application/octet-stream"
+    }
     
     with open(image_path, "rb") as image_file:
-        files = {"image": image_file}  # Change from "file" to "image"
-        response = requests.post(MODEL_URL, headers=headers, files=files)
+        image_data = image_file.read()
+
+    response = requests.post(MODEL_URL, headers=headers, data=image_data)
     
     if response.status_code == 200:
-        return response.json()[0]['generated_text']
+        try:
+            return response.json()[0]['generated_text']
+        except (IndexError, KeyError):
+            return f"Error: Unexpected response format: {response.text}"
     else:
         return f"Error: {response.text}"