Update main.py
Browse files
main.py
CHANGED
@@ -52,10 +52,12 @@ def LLModel():
|
|
52 |
|
53 |
def VectorDatabase(categorie):
|
54 |
if categorie != "year":
|
55 |
-
os.environ['PINECONE_API_KEY'] =
|
|
|
56 |
index_name = "all-venus"
|
57 |
else:
|
58 |
-
os.environ['PINECONE_API_KEY'] =
|
|
|
59 |
index_name = "all-jdlp"
|
60 |
|
61 |
embeddings = HuggingFaceEmbeddings()
|
|
|
52 |
|
53 |
def VectorDatabase(categorie):
|
54 |
if categorie != "year":
|
55 |
+
os.environ['PINECONE_API_KEY'] = ""
|
56 |
+
os.environ['PINECONE_API_KEY'] = os.getenv['PINECONE_API_KEY']
|
57 |
index_name = "all-venus"
|
58 |
else:
|
59 |
+
os.environ['PINECONE_API_KEY'] = ""
|
60 |
+
os.environ['PINECONE_API_KEY'] = os.getenv['PINECONE_API_KEYJDLP']
|
61 |
index_name = "all-jdlp"
|
62 |
|
63 |
embeddings = HuggingFaceEmbeddings()
|