osiria commited on
Commit
9047898
1 Parent(s): cd62ef1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -20,10 +20,9 @@ from unidecode import unidecode
20
  import string
21
  import re
22
 
23
- auth_token = os.environ.get("NEXT_IT_TOKEN")
24
 
25
- tokenizer = AutoTokenizer.from_pretrained("osiria/blaze-it-ner", use_auth_token=auth_token)
26
- model = DistilBertForTokenClassification.from_pretrained("osiria/blaze-it-ner", num_labels = 5, use_auth_token=auth_token)
27
  device = torch.device("cpu")
28
  model = model.to(device)
29
  model.eval()
 
20
  import string
21
  import re
22
 
 
23
 
24
+ tokenizer = AutoTokenizer.from_pretrained("osiria/blaze-it-ner")
25
+ model = DistilBertForTokenClassification.from_pretrained("osiria/blaze-it-ner", num_labels = 5)
26
  device = torch.device("cpu")
27
  model = model.to(device)
28
  model.eval()