Lin0He
/

text-summary-gpt2-short

text-generation

text-generation-inference

Model card Files Files and versions Community

Lin0He commited on Dec 5, 2023

Commit

1eee62f

·

1 Parent(s): 5222e7b

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -72,7 +72,7 @@ def model_infer(model, tokenizer, review, max_length=300):
 class EndpointHandler():
     def __init__(self, path=""):
         # load model and tokenizer from path
-        self.tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
         self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")

 class EndpointHandler():
     def __init__(self, path=""):
         # load model and tokenizer from path
+        self.tokenizer = AutoTokenizer.from_pretrained("Lin0He/text-summary-gpt2-short")
         self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")