Update app.py

app.py

@@ -11,8 +11,7 @@ model_path = "ayoolaolafenwa/ChatLM"
 
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 
-model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code = True, device_map = "auto",
-torch_dtype=torch.bfloat16, load_in_8bit=True)
+model = AutoModelForCausalLM.from_pretrained(model_path, device_map = "auto", torch_dtype=torch.bfloat16, load_in_8bit=True)
 
 class StopOnTokens(StoppingCriteria):
     def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs) -> bool: