leonarb commited on
Commit
36bb738
·
verified ·
1 Parent(s): 209c1bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -18,6 +18,15 @@ os.environ["TORCH_HOME"] = cache_dir
18
  os.environ["OLMOCR_LOG_PATH"] = "/tmp/olmocr-pipeline-debug.log"
19
  os.makedirs(cache_dir, exist_ok=True)
20
 
 
 
 
 
 
 
 
 
 
21
  # Import olmocr pipeline after setting log path
22
  from olmocr.pipeline import PDFToTextOCR
23
  from olmocr.data.renderpdf import render_pdf_to_base64png
 
18
  os.environ["OLMOCR_LOG_PATH"] = "/tmp/olmocr-pipeline-debug.log"
19
  os.makedirs(cache_dir, exist_ok=True)
20
 
21
+ # Patch logging path before olmocr import
22
+ import logging
23
+ original_file_handler = logging.FileHandler
24
+ def safe_file_handler(filename, *args, **kwargs):
25
+ if filename == "olmocr-pipeline-debug.log":
26
+ filename = os.environ.get("OLMOCR_LOG_PATH", "/tmp/olmocr-pipeline-debug.log")
27
+ return original_file_handler(filename, *args, **kwargs)
28
+ logging.FileHandler = safe_file_handler
29
+
30
  # Import olmocr pipeline after setting log path
31
  from olmocr.pipeline import PDFToTextOCR
32
  from olmocr.data.renderpdf import render_pdf_to_base64png