Update app.py

app.py

@@ -3,10 +3,10 @@ from transformers import AutoModel, AutoTokenizer
 from PIL import Image
 import tempfile
 
-# Load the tokenizer and model
+# Load the tokenizer and model without GPU
 tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
 model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, use_safetensors=True, pad_token_id=tokenizer.eos_token_id)
-model = model.eval()  # Ensure model is set to evaluation mode
+model = model.eval()  # Set to evaluation mode
 
 # Define the OCR function with error handling
 def perform_ocr(image):