Reizendretail commited on
Commit
155a8d7
1 Parent(s): eeb7f4e

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -19,7 +19,7 @@ class EndpointHandler():
19
  """
20
  # get inputs
21
  system_prompt = data.pop("system_prompt")
22
- message = data.pop("message")
23
  conversation = []
24
  conversation.append({"role": "system", "content": system_prompt})
25
  conversation.append({"role": "user", "content": message})
 
19
  """
20
  # get inputs
21
  system_prompt = data.pop("system_prompt")
22
+ message = data.pop("inputs")
23
  conversation = []
24
  conversation.append({"role": "system", "content": system_prompt})
25
  conversation.append({"role": "user", "content": message})