Update handler.py

handler.py

@@ -15,7 +15,7 @@ class EndpointHandler():
         self.model = AutoGPTQForCausalLM.from_quantized(path, device_map="auto", use_safetensors=True)
         self.tokenizer = AutoTokenizer.from_pretrained(path)
 
-    def get_input_token_length(message: str) -> int:
+    def get_input_token_length(self, message: str) -> int:
         input_ids = self.tokenizer([message], return_tensors='np', add_special_tokens=False)['input_ids']
         return input_ids.shape[-1]