Update app.py

app.py

@@ -19,7 +19,8 @@ from PIL import Image, ImageDraw, ImageFont
 
 
 processor = AutoProcessor.from_pretrained("microsoft/layoutlmv3-base", apply_ocr=True)
-model = AutoModelForTokenClassification.from_pretrained("Theivaprakasham/layoutlmv3-finetuned-invoice")
+#model = AutoModelForTokenClassification.from_pretrained("Theivaprakasham/layoutlmv3-finetuned-invoice")
+model = AutoModelForTokenClassification.from_pretrained("SickBoy/layoutlm_documents")
 
 
 
@@ -66,7 +67,7 @@ def process_image(image):
     width, height = image.size
 
     # encode
-    encoding = processor(image, truncation=False, return_offsets_mapping=True, return_tensors="pt")
+    encoding = processor(image, truncation=True, return_offsets_mapping=True, return_tensors="pt")
     offset_mapping = encoding.pop('offset_mapping')
 
     # forward pass