Update app.py

app.py

@@ -31,9 +31,9 @@ data_ready = False  # Flag to check if extraction is complete
 lock = threading.Lock()  # Lock to manage concurrent access
 extracted_texts = {}
 os.environ["HF_HOME"] = os.path.join(TEMP_DIR, "cache")  #"/app/cache"
-ocr_tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=False)
+ocr_tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
 ocr_model = AutoModel.from_pretrained(
-    'ucaslcl/GOT-OCR2_0', trust_remote_code=False,
+    'ucaslcl/GOT-OCR2_0', trust_remote_code=True,
     low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True
 ).eval().cuda()