lchakkei commited on
Commit
09c547d
·
verified ·
1 Parent(s): 5000e49

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -157,7 +157,7 @@ class EndpointHandler():
157
  )
158
 
159
  # Redact an answer
160
- answer = self.READER_LLM(final_prompt)[0]["generated_text"]
161
 
162
  return answer
163
 
 
157
  )
158
 
159
  # Redact an answer
160
+ answer = self.READER_LLM(final_prompt)[0]
161
 
162
  return answer
163