Update worker.py

worker.py

@@ -8,6 +8,9 @@ from langchain_community.vectorstores import Chroma
 from langchain_community.llms import HuggingFacePipeline
 from transformers import pipeline
 
+# Set a writable cache directory
+os.environ["HF_HOME"] = "./huggingface_cache"
+
 # Check for GPU availability
 DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"