Update files/functions.py

files/functions.py

@@ -59,9 +59,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-with-DocLayNet-base-at-linelevel-ml384"
 
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForTokenClassification.from_pretrained(model_id);
-model.to(device);
+API_TOKEN = "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"
+tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=API_TOKEN)
+model = AutoModelForTokenClassification.from_pretrained(model_id, use_auth_token=API_TOKEN);
+model.to(device);;
 
 # get labels
 id2label = model.config.id2label