Update app.py

app.py

@@ -32,7 +32,7 @@ def configure_llama_model():
     return llm
 
 def configure_embeddings():
-    embed_model = HuggingFaceEmbeddings(model_name="ggrn/e5-small-v2")
+    embed_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
     return embed_model
 
 
@@ -51,8 +51,9 @@ def initialize_vector_store_index(data_path, service_context):
         index = pickle.load(f)
     #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
+    documents = SimpleDirectoryReader("/content/Data/").load_data()
+    index = VectorStoreIndex.from_documents(documents, service_context=service_context)
+    return index
 
 def main():
     st.title("Cloudflare RAG")
@@ -68,10 +69,10 @@ def main():
 
     if user_input:
         # Generate response
-        docs = index.similarity_search(user_input)
-
+        query_engine = index.as_query_engine()
+        response = query_engine.query(user_input)
         # Display response
-        st.text_area("ChatGPT Response:", docs[0], height=100)
+        st.text_area("ChatGPT Response:", response, height=400)
 
 if __name__ == "__main__":
     main()