Update app.py

app.py

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