Update handler.py

handler.py

@@ -32,8 +32,10 @@ class EndpointHandler:
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         try:
-            output = self.process_json_input(data)
-            return output
+            if "inputs" in data:  # Check if data is in Hugging Face JSON format
+                return self.process_hf_input(data)
+            else:
+                return self.process_json_input(data)
         except ValueError as e:
             return {"error": str(e)}
         except Exception as e: