Update app.py

app.py

@@ -55,10 +55,16 @@ os.system('python -m pip install --upgrade pip')
 import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-from transformers import LayoutLMv2ForTokenClassification # LayoutXLMTokenizerFast, 
-# model
+# model 1
+from transformers import AutoTokenizer, AutoModelForTokenClassification
 model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-with-DocLayNet-base-at-linelevel-ml384"
-# tokenizer = LayoutXLMTokenizerFast.from_pretrained(model_id)
+tokenizer1 = AutoTokenizer.from_pretrained(model_id)
+model1 = AutoModelForTokenClassification.from_pretrained(model_id);
+model1.to(device);
+
+# model 2
+from transformers import LayoutLMv2ForTokenClassification # LayoutXLMTokenizerFast, 
+model_id = "pierreguillou/layout-xlm-base-finetuned-with-DocLayNet-base-at-linelevel-ml384"
 model2 = LayoutLMv2ForTokenClassification.from_pretrained(model_id);
 model2.to(device);
 
@@ -68,7 +74,8 @@ feature_extractor = LayoutLMv2FeatureExtractor(apply_ocr=False)
 
 # tokenizer
 from transformers import AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
+tokenizer_id = "xlm-roberta-base"
+tokenizer2 = AutoTokenizer.from_pretrained(tokenizer_id)
 
 # APP outputs 
 def app_outputs(uploaded_pdf):