Shakeel401 commited on
Commit
f877dec
·
verified ·
1 Parent(s): d9fa90e

Update services/qdrant_tool.py

Browse files
Files changed (1) hide show
  1. services/qdrant_tool.py +2 -3
services/qdrant_tool.py CHANGED
@@ -9,13 +9,12 @@ load_dotenv()
9
  qdrant_url = os.getenv("QDRANT_URL")
10
  qdrant_api = os.getenv("QDRANT_API_KEY")
11
 
12
- # ✅ Create writable cache directory for Hugging Face Spaces
13
- os.makedirs("./hf_cache", exist_ok=True)
14
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
15
  os.environ["HF_HOME"] = "/tmp/hf_cache"
16
  os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
17
 
18
- # ✅ Load embedding model from Hugging Face (with proper cache folder)
19
  model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="/tmp/hf_cache")
20
 
21
  # ✅ Connect to Qdrant Cloud
 
9
  qdrant_url = os.getenv("QDRANT_URL")
10
  qdrant_api = os.getenv("QDRANT_API_KEY")
11
 
12
+ # ✅ Set cache paths to Hugging Face writable directory
 
13
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
14
  os.environ["HF_HOME"] = "/tmp/hf_cache"
15
  os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
16
 
17
+ # ✅ Load embedding model
18
  model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder="/tmp/hf_cache")
19
 
20
  # ✅ Connect to Qdrant Cloud