Update files/functions.py

files/functions.py

@@ -57,11 +57,10 @@ sep_box = cls_box
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-DocLayNet-base_paragraphs_ml512-v4"
+model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-DocLayNet-base_paragraphs_ml512"
 
-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);
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForTokenClassification.from_pretrained(model_id);
 model.to(device);
 
 # get labels