viboognesh commited on
Commit
45a8816
·
verified ·
1 Parent(s): f5ee5a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -11,6 +11,8 @@ from langchain_openai import ChatOpenAI
11
  from langchain.memory import ConversationBufferMemory
12
  from langchain.chains import ConversationalRetrievalChain
13
 
 
 
14
  def getpdfdoc():
15
  with st.spinner("Loading PDF..."):
16
  filename = '48lawsofpower.pdf'
@@ -51,6 +53,8 @@ def get_text_chunks(text):
51
 
52
  def get_vectorstore(text_chunks):
53
  with st.spinner("Creating vectorstore..."):
 
 
54
  metadatas = [{"source": f"{i}-pl"} for i in range(len(text_chunks))]
55
  embeddings = OpenAIEmbeddings()
56
  vectorstore = Chroma.from_texts(texts=text_chunks, embedding=embeddings, persist_directory="./data/vectorstore", metadatas=metadatas)
 
11
  from langchain.memory import ConversationBufferMemory
12
  from langchain.chains import ConversationalRetrievalChain
13
 
14
+ import time
15
+
16
  def getpdfdoc():
17
  with st.spinner("Loading PDF..."):
18
  filename = '48lawsofpower.pdf'
 
53
 
54
  def get_vectorstore(text_chunks):
55
  with st.spinner("Creating vectorstore..."):
56
+ st.markdown("Creating vector store")
57
+ time.sleep(10)
58
  metadatas = [{"source": f"{i}-pl"} for i in range(len(text_chunks))]
59
  embeddings = OpenAIEmbeddings()
60
  vectorstore = Chroma.from_texts(texts=text_chunks, embedding=embeddings, persist_directory="./data/vectorstore", metadatas=metadatas)