Prajith04 commited on
Commit
ae93502
·
verified ·
1 Parent(s): 3f05777

Update utils/vector_search.py

Browse files
Files changed (1) hide show
  1. utils/vector_search.py +3 -1
utils/vector_search.py CHANGED
@@ -15,7 +15,9 @@ qdrant_client = QdrantClient(
15
  )
16
 
17
  # Initialize SentenceTransformer model
18
- model = SentenceTransformer("all-MiniLM-L6-v2")
 
 
19
 
20
  # Collection name (must match the one used during upsert)
21
  collection_name = "symptoms"
 
15
  )
16
 
17
  # Initialize SentenceTransformer model
18
+ cache_dir = os.environ.get("MODEL_CACHE_DIR", "/app/cache") # Fallback to /app/cache
19
+ os.makedirs(cache_dir, exist_ok=True)
20
+ model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder=cache_dir)
21
 
22
  # Collection name (must match the one used during upsert)
23
  collection_name = "symptoms"