Rename handler.py to pipeline.py

handler.py → pipeline.py

@@ -69,7 +69,7 @@ def model_infer(model, tokenizer, review, max_length=300):
         result_text.append(tokenizer.decode(result)) 
     return sorted(result_text, key=len)[3]
 
-class EndpointHandler():
+class PreTrainedPipeline():
     def __init__(self, path=""):
         # load model and tokenizer from path
         self.tokenizer = AutoTokenizer.from_pretrained("Lin0He/text-summary-gpt2-short")