dstars commited on
Commit
8b388d7
·
1 Parent(s): bc017bc

依赖冲突太麻烦了

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +2 -2
  2. app.py +2 -2
.ipynb_checkpoints/app-checkpoint.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="/root/model/sentence-transformer"
31
  )
32
  Settings.embed_model = embed_model
33
 
34
  #用初始化llm
35
  Settings.llm = llm
36
 
37
- documents = SimpleDirectoryReader("/root/llamaindex_demo/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-transformers/paraphrase-multilingual-MiniLM-L12-v2"
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
 
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="/root/model/sentence-transformer"
31
  )
32
  Settings.embed_model = embed_model
33
 
34
  #用初始化llm
35
  Settings.llm = llm
36
 
37
- documents = SimpleDirectoryReader("/root/llamaindex_demo/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-transformers/paraphrase-multilingual-MiniLM-L12-v2"
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