fix inputs

handler.py

@@ -32,9 +32,10 @@ class EndpointHandler:
             A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
-        prompt = data.pop("prompt", data)
+        inputs = data.pop("inputs", data)
         # get additional date field0
-        image_url = data.pop("files", None)[-1]['path']
+        prompt = inputs.pop("prompt", None)[-1]['path']
+        image_url = inputs.pop("files", None)[-1]['path']
 
         print(image_url)
         print(prompt)

test_inference_endpoint.py

@@ -0,0 +1,46 @@
+import os
+import requests
+from dotenv import load_dotenv
+
+
+load_dotenv()
+
+# global
+TOKEN =os.environ.get('HF_ACCESS_TOKEN')
+URL = "https://ojcld767sjuhadix.us-east-1.aws.endpoints.huggingface.cloud"
+headers = {"Authorization": f"Bearer {TOKEN}"}
+
+# prepare sample payload
+prompt = "Can you describe this picture focusing on specifics visual artifacts and ambiance (objects, colors, person, athmosphere..). Please stay concise only output keywords and concepts detected."
+files = [
+    {"path": "https://media.rolex.com/image/upload/q_auto/f_auto/c_limit,w_2440/v1708384234/rolexcom/about-rolex/hub/about-rolex-hub-cover-aca202310cw-0002-portrait"}
+]
+
+def call(payload):
+    response = requests.post(
+        URL,
+        headers=headers,
+        json=payload
+    )
+    return response.json()
+
+def test_local(payload):
+    from handler import EndpointHandler
+
+    # init handler
+    my_handler = EndpointHandler(path=".")
+    my_handler(payload)
+
+
+payload = {
+    "inputs": {
+        "prompt": prompt,
+        "files": files
+    }
+}
+
+# out_local = test_local(payload)
+# print(out_local)
+
+out = call(payload)
+print(out)