Update handler.py

handler.py

@@ -74,7 +74,7 @@ class LawLookup:
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = pipeline(model="amy011872/LawToken-0.5B-a2", device=0, torch_dtype=torch.float16)
+        self.pipeline = pipeline(model="amy011872/LawToken-7B-a2", device=0, torch_dtype=torch.float16)
         self.model = self.pipeline.model
         self.tokenizer = self.pipeline.tokenizer
         self.law_lookup = LawLookup('ChLaw.json')