osiria commited on
Commit
92fc2d8
1 Parent(s): 68b7cd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,8 +22,8 @@ import re
22
 
23
  auth_token = os.environ.get("AUTH_TOKEN")
24
 
25
- tokenizer = AutoTokenizer.from_pretrained("osiria/bert-base-cased-ner-en", use_auth_token=auth_token)
26
- model = BertForTokenClassification.from_pretrained("osiria/bert-base-cased-ner-en", num_labels = 5, use_auth_token=auth_token)
27
  device = torch.device("cpu")
28
  model = model.to(device)
29
  model.eval()
 
22
 
23
  auth_token = os.environ.get("AUTH_TOKEN")
24
 
25
+ tokenizer = AutoTokenizer.from_pretrained("osiria/bert-base-cased-ner-en", token=auth_token)
26
+ model = BertForTokenClassification.from_pretrained("osiria/bert-base-cased-ner-en", num_labels = 5, token=auth_token)
27
  device = torch.device("cpu")
28
  model = model.to(device)
29
  model.eval()