Update handler.py

handler.py

@@ -16,12 +16,13 @@ class EndpointHandler():
         self.model.eval()
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        encoded_images = data.get("images")
+        input_data = data.get("inputs", {})
+        encoded_images = input_data.get("images")
         
         if not encoded_images:
             return {"captions": [], "error": "No images provided"} 
 
-        texts = data.get("texts", ["a photography of"] * len(encoded_images)) 
+        texts = input_data.get("texts", ["a photography of"] * len(encoded_images)) 
 
         try:
             raw_images = [Image.open(BytesIO(base64.b64decode(img))).convert("RGB") for img in encoded_images]
@@ -41,4 +42,4 @@ class EndpointHandler():
             return {"captions": captions}
         except Exception as e:
             print(f"Error during processing: {str(e)}")
-            return {"captions": [], "error": str(e)}
+            return {"captions": [], "error": str(e)}