Spaces:
Sleeping
Sleeping
wencheng256
commited on
Commit
•
b081b86
1
Parent(s):
d27b005
Update myrpunct/punctuate.py
Browse files- 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 = "
|
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
|