skadio commited on
Commit
3bfb19c
·
1 Parent(s): a7d8fd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def load_models():
55
  model_args.use_cuda = False
56
  model_args.use_multiprocessing = False
57
  model = NERModel("roberta",
58
- "skadio/ner4opt-roberta",
59
  labels=custom_labels,
60
  use_cuda=False,
61
  args=model_args)
 
55
  model_args.use_cuda = False
56
  model_args.use_multiprocessing = False
57
  model = NERModel("roberta",
58
+ "skadio/ner4opt-roberta-v2",
59
  labels=custom_labels,
60
  use_cuda=False,
61
  args=model_args)