Update files/functions.py

files/functions.py

@@ -57,7 +57,7 @@ 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"
+model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-with-DocLayNet-base-at-paragraphlevel-ml512"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForTokenClassification.from_pretrained(model_id);