snrspeaks commited on
Commit
a64f5a9
·
1 Parent(s): b295741

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -14
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\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}))
 
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}))