Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ model.load_state_dict(torch.load(PATH + 'model_state_dict.pt',
|
|
14 |
map_location=DEVICE))
|
15 |
model.eval()
|
16 |
|
17 |
-
tokenizer = BertTokenizer.from_pretrained('
|
18 |
|
19 |
|
20 |
def evaluate(text) :
|
|
|
14 |
map_location=DEVICE))
|
15 |
model.eval()
|
16 |
|
17 |
+
tokenizer = BertTokenizer.from_pretrained('dslim/bert-base-NER-uncased')
|
18 |
|
19 |
|
20 |
def evaluate(text) :
|