Fix cache issues.

app.py

@@ -1,3 +1,13 @@
+import os
+
+# Force Hugging Face and Torch to use writable directories
+os.environ["TRANSFORMERS_CACHE"] = "/data/.cache"
+os.environ["HF_HOME"] = "/data/.cache"
+os.environ["TORCH_HOME"] = "/data/.cache"
+
+# Ensure the directory exists
+os.makedirs("/data/.cache", exist_ok=True)
+
 import gradio as gr
 import torch
 from PyPDF2 import PdfReader
@@ -10,8 +20,6 @@ from olmocr.prompts.anchor import get_anchor_text
 from ebooklib import epub
 import base64
 import tempfile
-import os
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"