Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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",
|
26 |
-
model = BertForTokenClassification.from_pretrained("osiria/bert-base-cased-ner-en", num_labels = 5,
|
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()
|