Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,25 +26,25 @@ def load():
|
|
| 26 |
embeddings = OpenAIEmbeddings()
|
| 27 |
llm = ChatOpenAI(temperature=0, model="gpt-4", streaming=True)
|
| 28 |
|
| 29 |
-
apple_2022_docs_store = FAISS.load_local(r'data
|
| 30 |
-
apple_2021_docs_store = FAISS.load_local(r'data
|
| 31 |
-
apple_2020_docs_store = FAISS.load_local(r'data
|
| 32 |
|
| 33 |
-
microsoft_2022_docs_store = FAISS.load_local(r'data
|
| 34 |
-
microsoft_2021_docs_store = FAISS.load_local(r'data
|
| 35 |
microsoft_2020_docs_store = FAISS.load_local(r'data\datastores\msft_2020', embeddings)
|
| 36 |
|
| 37 |
-
amazon_2022_docs_store = FAISS.load_local(r'data
|
| 38 |
-
amazon_2021_docs_store = FAISS.load_local(r'data
|
| 39 |
-
amazon_2020_docs_store = FAISS.load_local(r'data
|
| 40 |
|
| 41 |
-
alphabet_2022_docs_store = FAISS.load_local(r'data
|
| 42 |
-
alphabet_2021_docs_store = FAISS.load_local(r'data
|
| 43 |
-
alphabet_2020_docs_store = FAISS.load_local(r'data
|
| 44 |
|
| 45 |
-
meta_2022_docs_store = FAISS.load_local(r'data
|
| 46 |
-
meta_2021_docs_store = FAISS.load_local(r'data
|
| 47 |
-
meta_2020_docs_store = FAISS.load_local(r'data
|
| 48 |
|
| 49 |
|
| 50 |
apple_2022_qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=apple_2022_docs_store.as_retriever(search_kwargs={'k':5}))
|
|
|
|
| 26 |
embeddings = OpenAIEmbeddings()
|
| 27 |
llm = ChatOpenAI(temperature=0, model="gpt-4", streaming=True)
|
| 28 |
|
| 29 |
+
apple_2022_docs_store = FAISS.load_local(r'data/datastores/apple_2022', embeddings)
|
| 30 |
+
apple_2021_docs_store = FAISS.load_local(r'data/datastores/apple_2021', embeddings)
|
| 31 |
+
apple_2020_docs_store = FAISS.load_local(r'data/datastores/apple_2020', embeddings)
|
| 32 |
|
| 33 |
+
microsoft_2022_docs_store = FAISS.load_local(r'data/datastores/msft_2022', embeddings)
|
| 34 |
+
microsoft_2021_docs_store = FAISS.load_local(r'data/datastores/msft_2021', embeddings)
|
| 35 |
microsoft_2020_docs_store = FAISS.load_local(r'data\datastores\msft_2020', embeddings)
|
| 36 |
|
| 37 |
+
amazon_2022_docs_store = FAISS.load_local(r'data/datastores/amzn_2022', embeddings)
|
| 38 |
+
amazon_2021_docs_store = FAISS.load_local(r'data/datastores/amzn_2021', embeddings)
|
| 39 |
+
amazon_2020_docs_store = FAISS.load_local(r'data/datastores/amzn_2020', embeddings)
|
| 40 |
|
| 41 |
+
alphabet_2022_docs_store = FAISS.load_local(r'data/datastores/alphbt_2022', embeddings)
|
| 42 |
+
alphabet_2021_docs_store = FAISS.load_local(r'data/datastores/alphbt_2021', embeddings)
|
| 43 |
+
alphabet_2020_docs_store = FAISS.load_local(r'data/datastores/alphbt_2020', embeddings)
|
| 44 |
|
| 45 |
+
meta_2022_docs_store = FAISS.load_local(r'data/datastores/meta_2022', embeddings)
|
| 46 |
+
meta_2021_docs_store = FAISS.load_local(r'data/datastores/meta_2021', embeddings)
|
| 47 |
+
meta_2020_docs_store = FAISS.load_local(r'data/datastores/meta_2020', embeddings)
|
| 48 |
|
| 49 |
|
| 50 |
apple_2022_qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=apple_2022_docs_store.as_retriever(search_kwargs={'k':5}))
|