Update app.py

app.py

@@ -8,12 +8,13 @@ os.system('pip install -q pytesseract')
 
 import gradio as gr
 import numpy as np
-from transformers import LayoutLMv3Processor, LayoutLMv3ForTokenClassification
+
 from datasets import load_dataset
 from PIL import Image, ImageDraw, ImageFont
 
-processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv3-base-uncased")
-model = LayoutLMv2ForTokenClassification.from_pretrained("nielsr/layoutlmv3-finetuned-funsd")
+from transformers import AutoProcessor, AutoModelForTokenClassification
+processor = AutoProcessor.from_pretrained("Ammar-alhaj-ali/LayoutLMv3-Fine-Tuning-FUNSD")
+model = AutoModelForTokenClassification.from_pretrained("Ammar-alhaj-ali/LayoutLMv3-Fine-Tuning-FUNSD")
 
 # load image example
 dataset = load_dataset("nielsr/funsd", split="test")