RASMUS commited on
Commit
8a0d941
·
verified ·
1 Parent(s): d7c6080

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -2
handler.py CHANGED
@@ -33,8 +33,10 @@ class EndpointHandler():
33
  """
34
  print(data)
35
  try:
36
- context = data.pop("context",None)
37
- question = data.pop("question",None)
 
 
38
  messages = generate_rag_prompt_message(context, question)
39
 
40
  inputs = self.tokenizer(
 
33
  """
34
  print(data)
35
  try:
36
+ inputs = data.pop("inputs",None)
37
+ context = inputs["context"]
38
+ question = inputs["question"]
39
+
40
  messages = generate_rag_prompt_message(context, question)
41
 
42
  inputs = self.tokenizer(