Update handler.py

handler.py

@@ -25,7 +25,7 @@ class EndpointHandler():
 
     def __init__(self, path = ""):
         self.tokenizer = AutoTokenizer.from_pretrained(path)
-        self.model = torch.load("torch_model.pt")
+        self.model = torch.load(f"{path}/torch_model.pt")
     
     def response(self, result, user_name):
         result = result.rsplit("Alice Gate:", 1)[1].split(f"{user_name}:",1)[0].strip()