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-with-DocLayNet-base-at-linelevel-ml384"
+model_id = "NiamaLynn/lilt-roberta-DocLayNet-base_lines_ml256-v1"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForTokenClassification.from_pretrained(model_id);