Spaces:
Runtime error
Runtime error
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
|