Lin0He commited on
Commit
9d9d3b4
·
1 Parent(s): 77147a1

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -80,7 +80,7 @@ class EndpointHandler():
80
  self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
81
 
82
 
83
- def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
84
  # process input
85
  inputs = data.pop("inputs", data)
86
  # process input text
 
80
  self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
81
 
82
 
83
+ def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
84
  # process input
85
  inputs = data.pop("inputs", data)
86
  # process input text