kongs-git commited on
Commit
35674d4
·
1 Parent(s): c269a02

sent LlamaIndexRAG_InternLMAPI file

Browse files
app.py CHANGED
@@ -24,14 +24,14 @@ st.title("llamaindexRAG_lMAPI")
24
  @st.cache_resource
25
  def init_models():
26
  embed_model = HuggingFaceEmbedding(
27
- model_name="/root/LlamaIndexRAG_InternLMAPI/sentence-transformer"
28
  )
29
  Settings.embed_model = embed_model
30
 
31
  #用初始化llm
32
  Settings.llm = llm
33
 
34
- documents = SimpleDirectoryReader("/root/LlamaIndexRAG_InternLMAPI/data").load_data()
35
  index = VectorStoreIndex.from_documents(documents)
36
  query_engine = index.as_query_engine()
37
  return query_engine
 
24
  @st.cache_resource
25
  def init_models():
26
  embed_model = HuggingFaceEmbedding(
27
+ model_name="./paraphrase-multilingual-MiniLM-L12-v2"
28
  )
29
  Settings.embed_model = embed_model
30
 
31
  #用初始化llm
32
  Settings.llm = llm
33
 
34
+ documents = SimpleDirectoryReader("./data").load_data()
35
  index = VectorStoreIndex.from_documents(documents)
36
  query_engine = index.as_query_engine()
37
  return query_engine
paraphrase-multilingual-MiniLM-L12-v2 ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit e24d45fb8c210cedadb86b0dd70f3679bf216085
sentence-transformer DELETED
@@ -1 +0,0 @@
1
- Subproject commit 78fdff6834ee4d3ee67938f16cb3a6be582cd096