demo

README.md

@@ -6,6 +6,7 @@ colorTo: blue
 sdk: streamlit
 sdk_version: "0.0.1"
 app_file: app.py
+pinned: false
 ---
 
 # Llamaindex_RAG

app.py

@@ -26,14 +26,14 @@ st.title("llama_index_demo")
 @st.cache_resource
 def init_models():
     embed_model = HuggingFaceEmbedding(
-        model_name="/root/model/sentence-transformer"
+        model_name="/model"
     )
     Settings.embed_model = embed_model
 
     #用初始化llm
     Settings.llm = llm
 
-    documents = SimpleDirectoryReader("/root/llamaindex_demo/data").load_data()
+    documents = SimpleDirectoryReader("/data").load_data()
     index = VectorStoreIndex.from_documents(documents)
     query_engine = index.as_query_engine()