datacipen commited on
Commit
3696794
1 Parent(s): 263ee50

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -52,10 +52,12 @@ def LLModel():
52
 
53
  def VectorDatabase(categorie):
54
  if categorie != "year":
55
- os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
 
56
  index_name = "all-venus"
57
  else:
58
- os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEYJDLP']
 
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()