Commit
·
a32fe0f
1
Parent(s):
a9854aa
Update files/functions.py
Browse files- files/functions.py +1 -1
files/functions.py
CHANGED
|
@@ -62,7 +62,7 @@ model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-DocLayNet-base_paragra
|
|
| 62 |
API_TOKEN = "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"
|
| 63 |
tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=API_TOKEN)
|
| 64 |
model = AutoModelForTokenClassification.from_pretrained(model_id, use_auth_token=API_TOKEN);
|
| 65 |
-
model.to(device)
|
| 66 |
|
| 67 |
# get labels
|
| 68 |
id2label = model.config.id2label
|
|
|
|
| 62 |
API_TOKEN = "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"
|
| 63 |
tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=API_TOKEN)
|
| 64 |
model = AutoModelForTokenClassification.from_pretrained(model_id, use_auth_token=API_TOKEN);
|
| 65 |
+
model.to(device);
|
| 66 |
|
| 67 |
# get labels
|
| 68 |
id2label = model.config.id2label
|