Update app.py

app.py

@@ -10,9 +10,9 @@ MAX_INPUT_LIMIT = 3584
 
 MODEL_NAME = "THUDM/glm-4-9b-chat"
 
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True)
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
 
 GENERATE_CONFIG = dict(
     max_new_tokens=1536,