Update app.py

app.py

@@ -52,15 +52,15 @@ os.system('python -m pip install --upgrade pip')
 
 ## model / feature extractor / tokenizer
 
-from transformers import LayoutLMv2ForTokenClassification # LayoutXLMTokenizerFast, 
-
 import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+from transformers import LayoutLMv2ForTokenClassification # LayoutXLMTokenizerFast, 
 # model
+model_id = "pierreguillou/lilt-xlm-roberta-base-finetuned-with-DocLayNet-base-at-linelevel-ml384"
 # tokenizer = LayoutXLMTokenizerFast.from_pretrained(model_id)
-model = LayoutLMv2ForTokenClassification.from_pretrained(model_id);
-model.to(device);
+model2 = LayoutLMv2ForTokenClassification.from_pretrained(model_id);
+model2.to(device);
 
 # feature extractor
 from transformers import LayoutLMv2FeatureExtractor