Update handler.py
Browse files- 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
|