Hugging Face's logo

Spaces:
shibing624
/
pycorrector
Running

App Files Community
1

Update app.py

#1
by JoyceLiuZB - opened
base: refs/heads/main
←
from: refs/pr/1
Discussion Files changed
+2
-2
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):