Update app.py

app.py

@@ -1,8 +1,8 @@
 from langchain.chains import RetrievalQA
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain.callbacks.manager import CallbackManager
-from langchain_community.llms import Ollama
-from langchain_community.embeddings.ollama import OllamaEmbeddings
+#from langchain_community.llms import Ollama
+#from langchain_community.embeddings.ollama import OllamaEmbeddings
 from langchain_community.vectorstores import Chroma
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.document_loaders import PyPDFLoader
@@ -88,10 +88,8 @@ if uploaded_file is not None:
             all_splits = text_splitter.split_documents(data)
 
             # Create and persist the vector store
-            st.session_state.vectorstore = Chroma.from_documents(
-                documents=all_splits,
-                embedding=OllamaEmbeddings(model="mistral")
-            )
+            #st.session_state.vectorstore = Chroma.from_documents(documents=all_splits,embedding=OllamaEmbeddings(model="mistral"))
+            st.session_state.vectorstore = Chroma.from_documents(documents=all_splits,embedding=embeddings)
             st.session_state.vectorstore.persist()
 
     st.session_state.retriever = st.session_state.vectorstore.as_retriever()