Update handler.py

handler.py

@@ -71,10 +71,10 @@ def predict(text, model, tokenizer):
     #print("summary:", sorted(result_text, key=len)[3])
 
 class EndpointHandler():
-    def __init__(self, path=""):
+    def __init__(self, path="Lin0He/text-summary-gpt2-short"):
         # load model and tokenizer from path
-        self.tokenizer = AutoTokenizer.from_pretrained("Lin0He/text-summary-gpt2-short")
-        self.model = AutoModel.from_pretrained("Lin0He/text-summary-gpt2-short")
+        self.tokenizer = AutoTokenizer.from_pretrained(path)
+        self.model = AutoModel.from_pretrained(path)
 
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
@@ -82,7 +82,7 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         # process input text
         prediction = predict(inputs['text'], self.model, self.tokenizer)
-        return prediction #{"generated_text": prediction}
+        return {"generated_text": prediction}