Update handler.py

handler.py

@@ -18,8 +18,8 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         # get inputs
-        system_prompt = data.get("system_prompt")
-        message = data.get("message")
+        system_prompt = data.pop("system_prompt")
+        message = data.pop("message")
         conversation = []
         conversation.append({"role": "system", "content": system_prompt})
         conversation.append({"role": "user", "content": message})