Update app.py

app.py

@@ -1,21 +1,16 @@
 import streamlit as st
-
 from PIL import Image
 import re
-from transformers import AutoModel, AutoTokenizer
+from transformers import VisionEncoderDecoderModel, AutoProcessor
 
 st.set_page_config(page_title="OCR Application", page_icon="🖼️", layout="wide")
 device = "cpu"
 
 @st.cache_resource
-#def load_model():
-    #processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
-    #model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten', device_map='cpu')
-    #@st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, device_map='cpu')
-    model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cpu', use_safetensors=True)
-    processor=tokenizer
+    # Load processor and model for OCR
+    processor = AutoProcessor.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
+    model = VisionEncoderDecoderModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, device_map='cpu', low_cpu_mem_usage=True)
     return processor, model
 
 def extract_text(image, processor, model):