Update retriever.py

retriever.py

@@ -2,7 +2,13 @@ import chromadb
 from sentence_transformers import SentenceTransformer
 import os
 
-# Ensure the /tmp/data directory exists
+# Set Hugging Face cache directory to a writable location
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp/huggingface"
+
+# Ensure /tmp directories exist
+os.makedirs("/tmp/huggingface", exist_ok=True)
 os.makedirs("/tmp/data", exist_ok=True)
 
 # Initialize ChromaDB
@@ -10,8 +16,8 @@ os.makedirs("/tmp/data", exist_ok=True)
 chroma_client = chromadb.EphemeralClient()  # In-memory mode, resets on restart
 
 
-# Load embedding model
-embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
+# Load embedding model - previously: sentence-transformers/
+embed_model = SentenceTransformer("all-MiniLM-L6-v2")
 
 # Create collection (vector storage)
 collection = chroma_client.get_or_create_collection(name="documents")