Update app.py

app.py

@@ -1,21 +1,20 @@
 import streamlit as st
-from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
+from llama_index.core import VectorStoreIndex, Settings
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.llms.huggingface import HuggingFaceLLM
+from llama_index.core.readers import SimpleReader
 
 st.set_page_config(page_title="llama_index_demo", page_icon="🦜🔗")
 st.title("llama_index_demo")
 
-# Initialize models from Hugging Face Hub
+# Initialize models
 @st.cache_resource
 def init_models():
-    # Embeddings model
     embed_model = HuggingFaceEmbedding(
-        model_name="sentence-transformers/all-mpnet-base-v2" 
+        model_name="sentence-transformers/all-mpnet-base-v2"
     )
     Settings.embed_model = embed_model
 
-    # Language Model
     llm = HuggingFaceLLM(
         model_name="internlm/internlm2-chat-1_8b",
         tokenizer_name="internlm/internlm2-chat-1_8b",
@@ -24,8 +23,7 @@ def init_models():
     )
     Settings.llm = llm
 
-    # Load documents
-    documents = SimpleDirectoryReader("data.md").load_data() # Assuming data is in a "data" folder in your app directory
+    documents = SimpleReader().load_data(file_path="./data.md") #load data.md file
     index = VectorStoreIndex.from_documents(documents)
     query_engine = index.as_query_engine()