Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,13 +1,17 @@
 import streamlit as st
 import os
 import shutil  # Built-in, no need for pip
+
+# Set XDG_CACHE_HOME to writable /tmp before any imports that use it
+os.environ['XDG_CACHE_HOME'] = '/tmp'
+
 from transformers import pipeline
 
-# Set custom writable cache directory in /tmp to avoid permission issues
-os.environ['HF_HOME'] = '/tmp/huggingface_cache'
+# Set HF_HOME based on XDG (redirects cache to /tmp/huggingface)
+os.environ['HF_HOME'] = os.path.join(os.environ['XDG_CACHE_HOME'], 'huggingface')
 cache_dir = os.environ['HF_HOME']
 
-# Create cache dir if not exists ( /tmp is always writable in containers)
+# Create cache dir if not exists
 os.makedirs(cache_dir, exist_ok=True)
 
 # Clear any lock files in cache to fix interrupted downloads