halimb
/

depth-anything-small-hf

Depth Estimation

Inference Endpoints

Model card Files Files and versions Community

halimb commited on Feb 1

Commit

ead4cda

•

1 Parent(s): 9b01bc0

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -15,7 +15,7 @@ class EndpointHandler():
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        base64_image = data.pop("image",data)
         if base64_image is None:
             raise ValueError("No image provided")

     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        base64_image = data.pop("inputs",data)
         if base64_image is None:
             raise ValueError("No image provided")