Update src/streamlit_app.py

src/streamlit_app.py

@@ -10,6 +10,9 @@ from langchain_community.vectorstores import Chroma
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 from langchain_groq import ChatGroq
+from langchain_huggingface import HuggingFaceEmbeddings
+
+
 
 
 st.set_page_config(page_title="📘 PDF Q&A Generator", page_icon="🤖", layout="wide")
@@ -59,7 +62,8 @@ if "processed" not in st.session_state:
         splitter = RecursiveCharacterTextSplitter(chunk_size=600, chunk_overlap=100)
         texts = splitter.split_documents(documents)
 
-        embedding = OllamaEmbeddings(model="mxbai-embed-large")
+        #embedding = OllamaEmbeddings(model="mxbai-embed-large")
+        embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
         vectorstore = Chroma.from_documents(documents=texts, embedding=embedding)
         retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 4})