feat(remove model)
Browse files- app/config.py +1 -1
- download_model.py +0 -11
app/config.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
# Embedding model
|
| 4 |
-
EMBEDDING_MODEL = "
|
| 5 |
|
| 6 |
# Vector DB storage
|
| 7 |
CHROMA_DB_DIR = "/tmp/chroma_db"
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
# Embedding model
|
| 4 |
+
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
|
| 5 |
|
| 6 |
# Vector DB storage
|
| 7 |
CHROMA_DB_DIR = "/tmp/chroma_db"
|
download_model.py
DELETED
|
@@ -1,11 +0,0 @@
|
|
| 1 |
-
from sentence_transformers import SentenceTransformer
|
| 2 |
-
|
| 3 |
-
# Choose the model
|
| 4 |
-
model_name = "sentence-transformers/all-MiniLM-L6-v2"
|
| 5 |
-
|
| 6 |
-
# Load the model (this will download it)
|
| 7 |
-
model = SentenceTransformer(model_name)
|
| 8 |
-
|
| 9 |
-
# Save it locally
|
| 10 |
-
model.save("./embedding_model")
|
| 11 |
-
print("Model saved to ./embedding_model")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|