shibing624 commited on
Commit
cbd833c
1 Parent(s): 0643042

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -36,9 +36,9 @@ pip install -U pycorrector
36
  ```
37
 
38
  ```python
39
- from pycorrector.gpt.gpt_model import GptModel
40
- model = GptModel("chatglm", "THUDM/chatglm3-6b", peft_name="shibing624/chatglm3-6b-csc-chinese-lora")
41
- r = model.predict(["对下面文本纠错:\n少先队员因该为老人让坐。"])
42
  print(r) # ['少先队员应该为老人让座。']
43
  ```
44
 
 
36
  ```
37
 
38
  ```python
39
+ from pycorrector import GptCorrector
40
+ model = GptModel("THUDM/chatglm3-6b", "chatglm", peft_name="shibing624/chatglm3-6b-csc-chinese-lora")
41
+ r = model.correct_batch(["少先队员因该为老人让坐。"])
42
  print(r) # ['少先队员应该为老人让座。']
43
  ```
44