Update app.py

app.py

@@ -19,32 +19,32 @@ 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("microsoft/layoutlmv3-base")
 
 
 
 # load image example
-dataset = load_dataset("darentang/generated", split="test")
-Image.open(dataset[2]["image_path"]).convert("RGB").save("example1.png")
-Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
-Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
+# dataset = load_dataset("darentang/generated", split="test")
+# Image.open(dataset[2]["image_path"]).convert("RGB").save("example1.png")
+# Image.open(dataset[1]["image_path"]).convert("RGB").save("example2.png")
+# Image.open(dataset[0]["image_path"]).convert("RGB").save("example3.png")
 # define id2label, label2color
 labels = dataset.features['ner_tags'].feature.names
 id2label = {v: k for v, k in enumerate(labels)}
-label2color = {
-    "B-ABN": 'blue',
-    "B-BILLER": 'blue',
-    "B-BILLER_ADDRESS": 'green',
-    "B-BILLER_POST_CODE": 'orange',
-    "B-DUE_DATE": "blue",
-    "B-GST": 'green',
-    "B-INVOICE_DATE": 'violet',
-    "B-INVOICE_NUMBER": 'orange',
-    "B-SUBTOTAL": 'green',
-    "B-TOTAL": 'blue',
-    "I-BILLER_ADDRESS": 'blue',
-    "O": 'orange'
-  } 
+# label2color = {
+#     "B-ABN": 'blue',
+#     "B-BILLER": 'blue',
+#     "B-BILLER_ADDRESS": 'green',
+#     "B-BILLER_POST_CODE": 'orange',
+#     "B-DUE_DATE": "blue",
+#     "B-GST": 'green',
+#     "B-INVOICE_DATE": 'violet',
+#     "B-INVOICE_NUMBER": 'orange',
+#     "B-SUBTOTAL": 'green',
+#     "B-TOTAL": 'blue',
+#     "I-BILLER_ADDRESS": 'blue',
+#     "O": 'orange'
+#   } 
 
 def unnormalize_box(bbox, width, height):
      return [
@@ -92,11 +92,11 @@ def process_image(image):
     return image
 
 
-title = "Official Document Layout Scanner for a2i competition"
-description = "This is a web app for scanning official documents that will extract the layout from the documents automatically."
+title = "Demonstration of LayoutLmv3 for Document Layout Analysis"
+description = "Demo of Microsoft's LayoutLMv3 Document Foundation Model"
 
 
-examples =[['example1.png'],['example2.png'],['example3.png']]
+#examples =[['example1.png'],['example2.png'],['example3.png']]
 
 css = """.output_image, .input_image {height: 600px !important}"""