Update app.py

app.py

@@ -3,9 +3,9 @@ from transformers import AutoTokenizer, AutoModelForImageClassification
 from PIL import Image
 import torch
 
-# Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0')
-model = AutoModelForImageClassification.from_pretrained('ucaslcl/GOT-OCR2_0')
+# Load the model and tokenizer with trust_remote_code set to True
+tokenizer = AutoTokenizer.from_pretrained('stepfun-ai/GOT-OCR2_0', trust_remote_code=True)
+model = AutoModelForImageClassification.from_pretrained('stepfun-ai/GOT-OCR2_0', trust_remote_code=True)
 
 def perform_ocr(image):
     # Ensure the image is in the right format