wencheng256 commited on
Commit
b081b86
1 Parent(s): d27b005

Update myrpunct/punctuate.py

Browse files
Files changed (1) hide show
  1. myrpunct/punctuate.py +1 -1
myrpunct/punctuate.py CHANGED
@@ -14,7 +14,7 @@ class RestorePuncts:
14
  self.wrds_per_pred = wrds_per_pred
15
  self.overlap_wrds = 30
16
  self.valid_labels = ['OU', 'OO', '.O', '!O', ',O', '.U', '!U', ',U', ':O', ';O', ':U', "'O", '-O', '?O', '?U']
17
- self.model_hf = "wldmr/felflare-bert-restore-punctuation"
18
  self.model_args = NERArgs()
19
  self.model_args.silent = True
20
  self.model_args.max_seq_length = 512
 
14
  self.wrds_per_pred = wrds_per_pred
15
  self.overlap_wrds = 30
16
  self.valid_labels = ['OU', 'OO', '.O', '!O', ',O', '.U', '!U', ',U', ':O', ';O', ':U', "'O", '-O', '?O', '?U']
17
+ self.model_hf = "felflare/bert-restore-punctuation"
18
  self.model_args = NERArgs()
19
  self.model_args.silent = True
20
  self.model_args.max_seq_length = 512