Jeney commited on
Commit
db02aa7
1 Parent(s): 7807370

Output logits

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -26,4 +26,4 @@ class EndpointHandler:
26
  # postprocess the prediction
27
  logits = outputs.logits
28
  idx = logits.argmax(-1).item()
29
- return [{"best_answer": self.model.config.id2label[idx], "outputs":outputs}]
 
26
  # postprocess the prediction
27
  logits = outputs.logits
28
  idx = logits.argmax(-1).item()
29
+ return [{"best_answer": self.model.config.id2label[idx], "logits":outputs.logits}]