Update README.md

README.md

@@ -13,8 +13,8 @@ Usage:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 
-tokenizer = AutoTokenizer.from_pretrained("hiyouga/baichuan-7b-sft", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("hiyouga/baichuan-7b-sft", trust_remote_code=True).cuda()
+tokenizer = AutoTokenizer.from_pretrained("Hongbin37/CBT-LLM", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Hongbin37/CBT-LLM", trust_remote_code=True).cuda()
 streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
 query = "为什么生怕一点点事情做不好被人批评？做事情，别人告诉了我方法，但身体不会按方法来，非要折腾几遍，才发现别人告诉的方法和口诀，是最高效的；而且最近一个月，睡眠不好，整个白天都是无精打采的，每天活的很丧，知道自己的问题出在哪里，不晓得怎么去做出改变"