joonkim commited on
Commit
f372000
β€’
1 Parent(s): f9e20f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('facebook/bart-large-cnn')
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) :