Snowball commited on
Commit
2855cca
1 Parent(s): d1e5fc1

Update models/watermark_faster.py

Browse files
Files changed (1) hide show
  1. models/watermark_faster.py +2 -0
models/watermark_faster.py CHANGED
@@ -21,6 +21,8 @@ import paddle
21
  from jieba import posseg
22
  paddle.enable_static()
23
  import re
 
 
24
  def cut_sent(para):
25
  para = re.sub('([。!?\?])([^”’])', r'\1\n\2', para)
26
  para = re.sub('([。!?\?][”’])([^,。!?\?\n ])', r'\1\n\2', para)
 
21
  from jieba import posseg
22
  paddle.enable_static()
23
  import re
24
+ nltk.download('punkt')
25
+ nltk.download('stopwords')
26
  def cut_sent(para):
27
  para = re.sub('([。!?\?])([^”’])', r'\1\n\2', para)
28
  para = re.sub('([。!?\?][”’])([^,。!?\?\n ])', r'\1\n\2', para)