Update handler.py
Browse files- handler.py +3 -3
handler.py
CHANGED
@@ -74,10 +74,10 @@ def predict(text, model, tokenizer):
|
|
74 |
#print("summary:", sorted(result_text, key=len)[3])
|
75 |
|
76 |
class EndpointHandler():
|
77 |
-
def __init__(self, path="
|
78 |
# load model and tokenizer from path
|
79 |
-
self.tokenizer = AutoTokenizer.from_pretrained(
|
80 |
-
self.model = AutoModelForCausalLM.from_pretrained(
|
81 |
|
82 |
|
83 |
def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
|
|
|
74 |
#print("summary:", sorted(result_text, key=len)[3])
|
75 |
|
76 |
class EndpointHandler():
|
77 |
+
def __init__(self, path=""):
|
78 |
# load model and tokenizer from path
|
79 |
+
self.tokenizer = AutoTokenizer.from_pretrained("Lin0He/text-summary-gpt2-short")
|
80 |
+
self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
|
81 |
|
82 |
|
83 |
def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
|