fix naming
Browse files
sbert_punc_case_ru/sbertpunccase.py
CHANGED
@@ -69,7 +69,7 @@ def decode_label(label, classes='all'):
|
|
69 |
return INVERSE_LABELS[label]
|
70 |
|
71 |
|
72 |
-
MODEL_REPO = "kontur-ai/
|
73 |
|
74 |
|
75 |
class SbertPuncCase(nn.Module):
|
@@ -118,7 +118,7 @@ class SbertPuncCase(nn.Module):
|
|
118 |
|
119 |
|
120 |
if __name__ == '__main__':
|
121 |
-
parser = argparse.ArgumentParser("Punctuation and case restoration model
|
122 |
parser.add_argument("-i", "--input", type=str, help="text to restore", default='sbert punc case расставляет точки запятые и знаки вопроса вам нравится')
|
123 |
parser.add_argument("-d", "--device", type=str, help="run model on cpu or gpu", choices=['cpu', 'cuda'], default='cpu')
|
124 |
args = parser.parse_args()
|
|
|
69 |
return INVERSE_LABELS[label]
|
70 |
|
71 |
|
72 |
+
MODEL_REPO = "kontur-ai/sbert_punc_case_ru"
|
73 |
|
74 |
|
75 |
class SbertPuncCase(nn.Module):
|
|
|
118 |
|
119 |
|
120 |
if __name__ == '__main__':
|
121 |
+
parser = argparse.ArgumentParser("Punctuation and case restoration model sbert_punc_case_ru")
|
122 |
parser.add_argument("-i", "--input", type=str, help="text to restore", default='sbert punc case расставляет точки запятые и знаки вопроса вам нравится')
|
123 |
parser.add_argument("-d", "--device", type=str, help="run model on cpu or gpu", choices=['cpu', 'cuda'], default='cpu')
|
124 |
args = parser.parse_args()
|