change model path

handler.py

@@ -10,10 +10,11 @@ from transformers import (
 class EndpointHandler():
     def __init__(self, path=""):
         # Load model from HuggingFace Hub
+        self.model_path = path + "/" + "kobartbasekosummary.pt"
         config = BartConfig.from_pretrained("hyunwoongko/kobart")
         self.model = BartForConditionalGeneration(config).eval().to('cpu')
         self.model.model.load_state_dict(torch.load(
-            path,
+            self.model_path,
             map_location='cpu',
         ))
         self.tokenizer = PreTrainedTokenizerFast.from_pretrained("hyunwoongko/kobart")
@@ -43,7 +44,7 @@ class EndpointHandler():
         else:
             texts = dataPop
 
-        tokenized = tokenize(tokenizer, texts)
+        tokenized = self.tokenize(tokenizer, texts)
         input_ids = tokenized["input_ids"]
         attention_mask = tokenized["attention_mask"]
 
@@ -92,6 +93,7 @@ class EndpointHandler():
         return {"summarization": summ_result}
 
     def tokenize(
+            self,
             tokenizer,
             texts: List[str],
             max_len: int = 1024,
@@ -114,9 +116,9 @@ class EndpointHandler():
             # result + <eos>
         )
 
-        return add_bos_eos_tokens(tokenizer, tokens, eos_list)
+        return self.add_bos_eos_tokens(tokenizer, tokens, eos_list)
 
-    def add_bos_eos_tokens(tokenizer, tokens, eos_list):
+    def add_bos_eos_tokens(self, tokenizer, tokens, eos_list):
         input_ids = tokens["input_ids"]
         attention_mask = tokens["attention_mask"]
         token_added_ids, token_added_masks = [], []