Update app.py

app.py

@@ -8,7 +8,7 @@ import sentencepiece
 
 # Load the model and tokenizer using transformers
 model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-34B-200K", device_map="auto", torch_dtype="auto", trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained("Tonic/YiTonic", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("./")
 
 def run(message, chat_history, system_prompt, max_new_tokens=1024, temperature=0.3, top_p=0.9, top_k=50):
     prompt = get_prompt(message, chat_history, system_prompt)