Update app.py

app.py

@@ -1,24 +1,19 @@
 import os
 os.system('pip install git+https://github.com/huggingface/transformers.git --upgrade')
-os.system('pip install pyyaml==5.1')
 # workaround: install old version of pytorch since detectron2 hasn't released packages for pytorch 1.9 (issue: https://github.com/facebookresearch/detectron2/issues/3158)
 os.system('pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html')
 
-# install detectron2 that matches pytorch 1.8
-# See https://detectron2.readthedocs.io/tutorials/install.html for instructions
-os.system('pip install -q detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html')
-
 ## install PyTesseract
 os.system('pip install -q pytesseract')
 
 import gradio as gr
 import numpy as np
-from transformers import LayoutLMv2Processor, LayoutLMv2ForTokenClassification
+from transformers import LayoutLMv3Processor, LayoutLMv3ForTokenClassification
 from datasets import load_dataset
 from PIL import Image, ImageDraw, ImageFont
 
-processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv2-base-uncased")
-model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv2-finetuned-funsd")
+processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv3-base-uncased")
+model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-funsd")
 
 # load image example
 dataset = load_dataset("nielsr/funsd", split="test")