Upload folder using huggingface_hub

handler.py

@@ -58,8 +58,15 @@ class EndpointHandler:
             return False
 
     def _decode_image(self, data):
-        if isinstance(data, str):
-            image_bytes = base64.b64decode(data)
-        else:
-            image_bytes = data
-        return Image.open(io.BytesIO(image_bytes)).convert("RGB")
+        try:
+            if isinstance(data, str):
+                image_bytes = base64.b64decode(data)
+            else:
+                image_bytes = data
+            img = Image.open(io.BytesIO(image_bytes))
+            # Ensure loaded
+            img.load()
+            return img.convert("RGB")
+        except Exception as e:
+            print(f"Image decode failed: {e}")
+            raise ValueError(f"Invalid image data: {e}")