re-mind
/

aimv2-large-patch14-native-embedding-inference

Model card Files Files and versions

amaye15 commited on Jan 23

Commit

0366c0a

·

1 Parent(s): 7e994bd

Debug - Handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -104,7 +104,7 @@ class EndpointHandler:
         Returns:
             Dict[str, Any]: Dictionary containing encoded embeddings or error messages.
         """
-        images_data = data.get("images", [])
         if not images_data:
             return {"error": "No image data provided."}

         Returns:
             Dict[str, Any]: Dictionary containing encoded embeddings or error messages.
         """
+        images_data = data.get("inputs", [])
         if not images_data:
             return {"error": "No image data provided."}