Update app.py

app.py

@@ -36,6 +36,18 @@ from functions import *
 # update pip
 os.system('python -m pip install --upgrade pip')
 
+# model
+from transformers import AutoTokenizer, AutoModelForTokenClassification
+
+import torch
+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"
+
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForTokenClassification.from_pretrained(model_id);
+model.to(device);
+
 # APP outputs 
 def app_outputs(uploaded_pdf):
     filename, msg, images = pdf_to_images(uploaded_pdf)