Spaces:
Running
Running
Update app.py
Browse files
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"
|
26 |
-
model = DistilBertForTokenClassification.from_pretrained("osiria/blaze-it-ner", num_labels = 5
|
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()
|