Spaces:
Sleeping
Sleeping
Update vectordb_utils.py
Browse files- vectordb_utils.py +1 -0
vectordb_utils.py
CHANGED
@@ -4,6 +4,7 @@ from qdrant_client import QdrantClient
|
|
4 |
from qdrant_client.models import VectorParams, Distance, PointStruct
|
5 |
from sentence_transformers import SentenceTransformer
|
6 |
import uuid
|
|
|
7 |
cache_dir = os.environ.get("MODEL_CACHE_DIR", "/app/cache") # Fallback to /app/cache
|
8 |
os.makedirs(cache_dir, exist_ok=True)
|
9 |
encoder =SentenceTransformer("all-MiniLM-L6-v2", cache_folder=cache_dir)
|
|
|
4 |
from qdrant_client.models import VectorParams, Distance, PointStruct
|
5 |
from sentence_transformers import SentenceTransformer
|
6 |
import uuid
|
7 |
+
import os
|
8 |
cache_dir = os.environ.get("MODEL_CACHE_DIR", "/app/cache") # Fallback to /app/cache
|
9 |
os.makedirs(cache_dir, exist_ok=True)
|
10 |
encoder =SentenceTransformer("all-MiniLM-L6-v2", cache_folder=cache_dir)
|