Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,10 +20,8 @@ from unidecode import unidecode
|
|
20 |
import string
|
21 |
import re
|
22 |
|
23 |
-
|
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()
|