Update app.py

app.py

@@ -7,7 +7,7 @@ from llama_index.llms.llama_utils import messages_to_prompt, completion_to_promp
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings
 from langchain.embeddings.huggingface import HuggingFaceEmbeddings
 from llama_index.embeddings import LangchainEmbedding
-
+from langchain.embeddings import SentenceTransformerEmbeddings
 
 
 # Set up logging
@@ -32,6 +32,9 @@ def configure_embeddings():
     embed_model = HuggingFaceEmbeddings(model_name="ggrn/e5-small-v2")
     return embed_model
 
+
+
+
 def configure_service_context(llm, embed_model):
     return ServiceContext.from_defaults(chunk_size=256, llm=llm, embed_model=embed_model)
 
@@ -42,9 +45,10 @@ def initialize_vector_store_index(data_path, service_context):
     # Load the index from a file
     with open('./index_file.pkl', 'rb') as f:
         index = pickle.load(f)
-    index = VectorStoreIndex.from_documents(documents, service_context=service_context)
-
-    return index
+    #index = VectorStoreIndex.from_documents(documents, service_context=service_context)
+    embeddings_2 = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
+    index2 = FAISS.from_documents(pages, embeddings_2)
+    return index2
 
 def main():
     st.title("Cloudflare RAG")
@@ -60,8 +64,7 @@ def main():
 
     if user_input:
         # Generate response
-        query_engine = index.as_query_engine()
-        response = query_engine.query(user_input)
+        docs = index2.similarity_search(user_input)
 
         # Display response
         st.text_area("ChatGPT Response:", response, height=100)