Update app.py

app.py

@@ -4,9 +4,12 @@ from PIL import Image
 import gradio as gr
 import os
 
-# Load the OCR model and tokenizer, trust_remote_code=True allows custom model logic
-tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
-model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, 
+# Specify the revision ID you want to pin to
+revision_id = "your_revision_id_here"  # Replace with the actual revision ID
+
+# Load the OCR model and tokenizer with pinned revision
+tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', revision=revision_id, trust_remote_code=True)
+model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', revision=revision_id, trust_remote_code=True, 
                                   low_cpu_mem_usage=True, 
                                   pad_token_id=tokenizer.eos_token_id).eval()