Spaces:

shibing624
/

pycorrector

Running

shibing624 commited on Mar 27, 2023

Commit

70c38a6

1 Parent(s): be5922c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ import operator
 import torch
 from transformers import BertTokenizer, BertForMaskedLM
-tokenizer = BertTokenizer.from_pretrained("shibing624/bart4csc-base-chinese")
-model = BertForMaskedLM.from_pretrained("shibing624/bart4csc-base-chinese")
 def ai_text(text):

 import torch
 from transformers import BertTokenizer, BertForMaskedLM
+tokenizer = BertTokenizer.from_pretrained("shibing624/macbert4csc-base-chinese")
+model = BertForMaskedLM.from_pretrained("shibing624/macbert4csc-base-chinese")
 def ai_text(text):