Maki Nishikino commited on
Commit
59062a6
·
1 Parent(s): 41d8852

Change path

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,14 +25,14 @@ st.title("llama_index_demo")
25
  @st.cache_resource
26
  def init_models():
27
  embed_model = HuggingFaceEmbedding(
28
- model_name="/root/model/sentence-transformer"
29
  )
30
  Settings.embed_model = embed_model
31
 
32
  #用初始化llm
33
  Settings.llm = llm
34
 
35
- documents = SimpleDirectoryReader("/root/llamaindex_demo/data").load_data()
36
  index = VectorStoreIndex.from_documents(documents)
37
  query_engine = index.as_query_engine()
38
 
 
25
  @st.cache_resource
26
  def init_models():
27
  embed_model = HuggingFaceEmbedding(
28
+ model_name="./paraphrase-multilingual-MiniLM-L12-v2"
29
  )
30
  Settings.embed_model = embed_model
31
 
32
  #用初始化llm
33
  Settings.llm = llm
34
 
35
+ documents = SimpleDirectoryReader("./data").load_data()
36
  index = VectorStoreIndex.from_documents(documents)
37
  query_engine = index.as_query_engine()
38