arun46 commited on
Commit
43facdb
·
verified ·
1 Parent(s): 2addb95

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -7,8 +7,11 @@ from utils.pdf_to_images import convert_pdf_to_images, convert_image_to_pil
7
  from utils.hunyuan_ocr import HunyuanOCR
8
  from utils.pdf_builder import build_searchable_pdf
9
 
 
 
 
10
  # Initialize OCR model
11
- ocr_model = HunyuanOCR()
12
 
13
  def process_document(file_obj):
14
  if file_obj is None:
 
7
  from utils.hunyuan_ocr import HunyuanOCR
8
  from utils.pdf_builder import build_searchable_pdf
9
 
10
+ # Get Hugging Face token from environment variables (for gated models)
11
+ hf_token = os.environ.get("HF_TOKEN")
12
+
13
  # Initialize OCR model
14
+ ocr_model = HunyuanOCR(token=hf_token)
15
 
16
  def process_document(file_obj):
17
  if file_obj is None: