Spaces:
Paused
Paused
Commit ·
1322f40
1
Parent(s): 4eaa670
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,6 +22,11 @@ from threading import Thread
|
|
| 22 |
# pinecone db index
|
| 23 |
index_name = "kellogg-markstrat"
|
| 24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
# set up OpenAI environment vars and embeddings
|
| 26 |
openai.api_key = os.environ.get("OPENAI_API_KEY")
|
| 27 |
embeddings = OpenAIEmbeddings()
|
|
@@ -75,12 +80,6 @@ def predict(message, model_type):
|
|
| 75 |
q = Queue()
|
| 76 |
job_done = object()
|
| 77 |
|
| 78 |
-
# pinecone DB init
|
| 79 |
-
pinecone.init(
|
| 80 |
-
api_key=os.getenv("PINECONE_API_KEY"), # find at app.pinecone.io
|
| 81 |
-
environment=os.getenv("PINECONE_ENV"), # next to api key in console
|
| 82 |
-
)
|
| 83 |
-
|
| 84 |
# load existing index
|
| 85 |
vectorsearch = Pinecone.from_existing_index(index_name, embeddings)
|
| 86 |
retriever = vectorsearch.as_retriever()
|
|
|
|
| 22 |
# pinecone db index
|
| 23 |
index_name = "kellogg-markstrat"
|
| 24 |
|
| 25 |
+
Pinecone.init(
|
| 26 |
+
api_key=os.getenv("PINECONE_API_KEY"), # find at app.pinecone.io
|
| 27 |
+
environment=os.getenv("PINECONE_ENV"), # next to api key in console
|
| 28 |
+
)
|
| 29 |
+
|
| 30 |
# set up OpenAI environment vars and embeddings
|
| 31 |
openai.api_key = os.environ.get("OPENAI_API_KEY")
|
| 32 |
embeddings = OpenAIEmbeddings()
|
|
|
|
| 80 |
q = Queue()
|
| 81 |
job_done = object()
|
| 82 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 83 |
# load existing index
|
| 84 |
vectorsearch = Pinecone.from_existing_index(index_name, embeddings)
|
| 85 |
retriever = vectorsearch.as_retriever()
|