remove req

handler.py

@@ -24,9 +24,12 @@ class EndpointHandler:
         # self.model = model
         # self.device = device
 
-    def __call__(self, data):
-        return data
-
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        inputs = data.get("inputs", "")
+        if not inputs:
+            return [{"error": "No inputs provided"}]
+        
+        return inputs
     # def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
     #     """
     #     data args:

test_inference_endpoint.py

@@ -19,7 +19,7 @@ def call(payload):
     response = requests.post(
         URL,
         headers=headers,
-        json=payload
+        json={'inputs': 'yo'}
     )
     print(response)
     return response.json()