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}))
|