change model directory
Browse files- src/conversation.py +1 -1
- src/vector_index.py +1 -1
src/conversation.py
CHANGED
@@ -16,7 +16,7 @@ class Conversation_RAG:
|
|
16 |
|
17 |
def get_vectordb(self):
|
18 |
index = pinecone.Index(os.environ.get("PINECONE_INDEX"))
|
19 |
-
embeddings = HuggingFaceEmbeddings(model_name=f"model/{model_name}")
|
20 |
vectordb = Pinecone(index, embeddings, "text")
|
21 |
|
22 |
return vectordb
|
|
|
16 |
|
17 |
def get_vectordb(self):
|
18 |
index = pinecone.Index(os.environ.get("PINECONE_INDEX"))
|
19 |
+
embeddings = HuggingFaceEmbeddings(model_name=f"./model/{model_name}")
|
20 |
vectordb = Pinecone(index, embeddings, "text")
|
21 |
|
22 |
return vectordb
|
src/vector_index.py
CHANGED
@@ -31,7 +31,7 @@ def create_vector_store_index(file_path):
|
|
31 |
|
32 |
index = pc.Index(os.environ.get("PINECONE_INDEX"))
|
33 |
|
34 |
-
embeddings = HuggingFaceEmbeddings(model_name=f"model/{model_name}")
|
35 |
|
36 |
batch_size = 32
|
37 |
|
|
|
31 |
|
32 |
index = pc.Index(os.environ.get("PINECONE_INDEX"))
|
33 |
|
34 |
+
embeddings = HuggingFaceEmbeddings(model_name=f"./model/{model_name}")
|
35 |
|
36 |
batch_size = 32
|
37 |
|