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
|