Update app.py

app.py

@@ -96,7 +96,8 @@ def rag_chatbot():
         llm_model = load_llm(api_key=api[:-2])
 
         # Create retriever using Chroma
-        retriever = Chroma.from_documents(documents, embedding=embedding_model, persist_directory=None).as_retriever()
+         # FAISS.from_documents(documents, embedding)
+        retriever = FAISS.from_documents(documents, embedding=embedding_model).as_retriever()
 
         # Ask a question
         question = st.text_input("Ask a question about your dataset:")