Update app.py
Browse files
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)
|