Update handler.py

handler.py

@@ -14,13 +14,11 @@ CONFIG = {
     'do_sample': True,
 }
 
-class EndpointHandler():
-    def __init__(self, 
-                 model_path: str = ".", 
-                 tokenizer_path: str = "."):
+class EndpointHandler:
+    def __init__(self, path: str = ""):
         
-        self.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
-        self.model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
+        self.tokenizer = AutoTokenizer.from_pretrained(path)
+        self.model = AutoModelForSeq2SeqLM.from_pretrained(path)
 
     def __call__(self, data: Dict[str, str]) -> Dict[str, str]: