Kareman commited on
Commit
3de283b
·
1 Parent(s): 01ad7ee

feat(remove model)

Browse files
Files changed (2) hide show
  1. app/config.py +1 -1
  2. download_model.py +0 -11
app/config.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
 
3
  # Embedding model
4
- EMBEDDING_MODEL = "./embedding_model" # path to local folder
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")