update handler.py

handler-test.py

@@ -0,0 +1,20 @@
+# pip install -r requirements.txt
+
+from handler import EndpointHandler
+
+# init handler
+my_handler = EndpointHandler(path=".")
+
+# prepare sample payload
+payload = {
+    "inputs": {
+        "question": "What's in the image?",
+        "image": "https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg"
+    }
+}
+
+# test the handler
+result = my_handler(payload)
+
+# show results
+print("result:", result)

handler.py

@@ -14,9 +14,10 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
-        image = "https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg"
-        question = data.pop("question", data)
-        # inputs = data.pop("inputs", data)
+        inputs = data.pop("inputs", data)
+        top_k = inputs.get("top_k", 10)
+        question = inputs.get("question", 'Whats the negative of "You provided a question"?')
+        image = inputs.get("image", "https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg")
 
         # run normal prediction
         prediction = self.pipeline(image, question, top_k=10)