Spaces:
Running
Running
Update app.py
Browse files
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
|