Update app.py

app.py

@@ -25,7 +25,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model_name = "01-ai/Yi-6B-200K"
 
-# tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
+tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 # tokenizer = YiTokenizer(vocab_file=model_name)
 model = transformers.AutoModelForCausalLM.from_pretrained(model_name,
     device_map=device,