Spaces:

shibing624
/

pycorrector

Running

JoyceLiuZB commited on Feb 1, 2023

Commit

7fa0404

•

1 Parent(s): 6110b96

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/macbert4csc-base-chinese")
-model = BertForMaskedLM.from_pretrained("shibing624/macbert4csc-base-chinese")
 def ai_text(text):

 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):