Update handler

.gitignore

@@ -0,0 +1,2 @@
+test_hf.py
+venv/

handler.py

@@ -14,8 +14,9 @@ class EndpointHandler():
       Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
-        image = data.pop("image")
-        prompt = data.pop("prompt")
+        inputs = data.pop("inputs")
+        image = inputs.pop("image")
+        prompt = inputs.pop("prompt")
 
         return [{
             "image": image,