Inference-APP-Document-Understanding-at-paragraphlevel-v1

Runtime error

pierreguillou commited on Feb 14, 2023

Commit

03f879f

1 Parent(s): e3657e8

Update files/functions.py

Files changed (1) hide show

files/functions.py CHANGED Viewed

@@ -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"
 API_TOKEN = "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"
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=API_TOKEN)

 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"
 API_TOKEN = "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"
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=API_TOKEN)