Update worker.py

worker.py

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