fix inputs

handler.py

@@ -34,8 +34,8 @@ class EndpointHandler:
         # get inputs
         inputs = data.pop("inputs", data)
         # get additional date field0
-        prompt = inputs.pop("prompt", None)[-1]['path']
-        image_url = inputs.pop("files", None)[-1]['path']
+        prompt = inputs.pop("prompt", None)
+        image_url = inputs.pop("image", None)
 
         print(image_url)
         print(prompt)

test_inference_endpoint.py

@@ -12,9 +12,7 @@ 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"}
-]
+image = "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(
@@ -35,7 +33,7 @@ def test_local(payload):
 payload = {
     "inputs": {
         "prompt": prompt,
-        "files": files
+        "image": image
     }
 }