Update handler.py

handler.py

@@ -2,10 +2,10 @@ from typing import Dict, List, Any
 from transformers import AutoTokenizer, AutoModel
 
 
-class EndpointHandler():
+class EndpointHandler:
     def __init__(self):
-        self.tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True)
-        self.model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True).half().cuda()
+        self.tokenizer = AutoTokenizer.from_pretrained("chatglm2-6b-int4", trust_remote_code=True)
+        self.model = AutoModel.from_pretrained("chatglm2-6b-int4", trust_remote_code=True).half().cuda()
         self.model = self.model.eval()