BlueDice commited on
Commit
82bd0a9
·
1 Parent(s): 91f9c84

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -10,7 +10,7 @@ class EndpointHandler():
10
  self.model = torch.load(f"{path}/torch_model.pt")
11
 
12
  def __call__(self, data):
13
- request_inputs = input_data.pop("inputs", data)
14
  template = request_inputs["template"]
15
  messages = request_inputs["messages"]
16
  char_name = request_inputs["char_name"]
 
10
  self.model = torch.load(f"{path}/torch_model.pt")
11
 
12
  def __call__(self, data):
13
+ request_inputs = data.pop("inputs", data)
14
  template = request_inputs["template"]
15
  messages = request_inputs["messages"]
16
  char_name = request_inputs["char_name"]