Spaces:
Runtime error
Runtime error
Update concat_vector_store_정리된.py
Browse files
concat_vector_store_정리된.py
CHANGED
@@ -10,7 +10,7 @@ FOLDER = "cleaned_pdfs" # Folder containing the cleaned PDFs
|
|
10 |
VECTOR_STORE_PATH = "vector_db"
|
11 |
|
12 |
# 1. Load the embedding model
|
13 |
-
def get_embeddings(model_name="
|
14 |
return E5Embeddings(
|
15 |
model_name=model_name,
|
16 |
model_kwargs={'device': device},
|
|
|
10 |
VECTOR_STORE_PATH = "vector_db"
|
11 |
|
12 |
# 1. Load the embedding model
|
13 |
+
def get_embeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", device="cuda"):
|
14 |
return E5Embeddings(
|
15 |
model_name=model_name,
|
16 |
model_kwargs={'device': device},
|