Hugging Face's logo

Spaces:
luckpunk
/
pycorrector
Runtime error

App Files Community
shibing624 JoyceLiuZB commited on
Commit
be5922c
·
1 Parent(s): 5d51016

Update app.py (#1)

Browse files

- Update app.py (7fa0404b7f1db2c213c9baa0ca053c4f6bd2a801)


Co-authored-by: JoyceLiu <[email protected]>

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ import operator
9
  import torch
10
  from transformers import BertTokenizer, BertForMaskedLM
11
 
12
- tokenizer = BertTokenizer.from_pretrained("shibing624/macbert4csc-base-chinese")
13
- model = BertForMaskedLM.from_pretrained("shibing624/macbert4csc-base-chinese")
14
 
15
 
16
  def ai_text(text):
 
9
  import torch
10
  from transformers import BertTokenizer, BertForMaskedLM
11
 
12
+ tokenizer = BertTokenizer.from_pretrained("shibing624/bart4csc-base-chinese")
13
+ model = BertForMaskedLM.from_pretrained("shibing624/bart4csc-base-chinese")
14
 
15
 
16
  def ai_text(text):