Spaces:
Sleeping
Sleeping
Update services/qdrant_tool.py
Browse files- 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 |
-
# ✅
|
| 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
|
| 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
|