Update app.py

app.py

@@ -46,7 +46,6 @@ if not torch.cuda.is_available():
 
 # if torch.cuda.is_available():
 model_id = "DAMO-NLP-SG/CLEX-7b-Chat-16K"
-# from CLEX import LlamaForCausalLM
 from transformers import AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)