osiria commited on
Commit
43c0e48
·
verified ·
1 Parent(s): c149e5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -20,10 +20,8 @@ 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/flare-it-ner", use_auth_token=auth_token)
26
- model = XLMRobertaForTokenClassification.from_pretrained("osiria/flare-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
+ tokenizer = AutoTokenizer.from_pretrained("osiria/flare-it-ner")
24
+ model = XLMRobertaForTokenClassification.from_pretrained("osiria/flare-it-ner", num_labels = 5)
 
 
25
  device = torch.device("cpu")
26
  model = model.to(device)
27
  model.eval()