penguin218 commited on
Commit
33cdcbb
·
1 Parent(s): 70b139f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,14 +27,14 @@ st.title("llama_index_demo")
27
  @st.cache_resource
28
  def init_models():
29
  embed_model = HuggingFaceEmbedding(
30
- model_name="sentence-transformer"
31
  )
32
  Settings.embed_model = embed_model
33
 
34
  #用初始化llm
35
  Settings.llm = llm
36
 
37
- documents = SimpleDirectoryReader("data").load_data()
38
  index = VectorStoreIndex.from_documents(documents)
39
  query_engine = index.as_query_engine()
40
 
 
27
  @st.cache_resource
28
  def init_models():
29
  embed_model = HuggingFaceEmbedding(
30
+ model_name="./sentence-transformer"
31
  )
32
  Settings.embed_model = embed_model
33
 
34
  #用初始化llm
35
  Settings.llm = llm
36
 
37
+ documents = SimpleDirectoryReader("./data").load_data()
38
  index = VectorStoreIndex.from_documents(documents)
39
  query_engine = index.as_query_engine()
40