Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,12 +6,12 @@ import os
 import numpy as np
 import chromadb
 from chromadb.utils import embedding_functions
+import tempfile  # ✅ For safe writable cache directory
 
 # Initialize session state
 if 'model' not in st.session_state:
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    cache_dir = "./clip_cache"
-    os.makedirs(cache_dir, exist_ok=True)
+    cache_dir = tempfile.gettempdir()  # ✅ Use temp dir instead of restricted local path
     try:
         model, preprocess = clip.load("ViT-B/32", device=device, download_root=cache_dir)
     except Exception as e:
@@ -38,7 +38,7 @@ if 'chroma_client' not in st.session_state:
         st.error(f"Failed to initialize ChromaDB: {e}")
         st.stop()
 
-# Load demo images only once
+# Load demo images once
 if not st.session_state.get("demo_images_loaded", False):
     demo_folder = "demo_images"
     if os.path.exists(demo_folder):