Almira commited on
Commit
f35bab7
1 Parent(s): fbfa992

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/sbert-punc-case-ru"
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 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()
 
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()