sfang32 commited on
Commit
432418d
·
verified ·
1 Parent(s): d6e8c88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -16,8 +16,6 @@ api_key = os.getenv('key_internlm')
16
  # model = "internlm/internlm2_5-7b-chat"
17
  # api_key = "请填写 API Key"
18
 
19
- path_model_embedding = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
20
- path_knowledge = "data_domain"
21
 
22
  llm =OpenAILike(
23
  model=model,
@@ -34,14 +32,14 @@ st.title("llama_index_demo")
34
  @st.cache_resource
35
  def init_models():
36
  embed_model = HuggingFaceEmbedding(
37
- model_name=path_model_embedding
38
  )
39
  Settings.embed_model = embed_model
40
 
41
  #用初始化llm
42
  Settings.llm = llm
43
 
44
- documents = SimpleDirectoryReader(path_knowledge).load_data()
45
  index = VectorStoreIndex.from_documents(documents)
46
  query_engine = index.as_query_engine()
47
 
 
16
  # model = "internlm/internlm2_5-7b-chat"
17
  # api_key = "请填写 API Key"
18
 
 
 
19
 
20
  llm =OpenAILike(
21
  model=model,
 
32
  @st.cache_resource
33
  def init_models():
34
  embed_model = HuggingFaceEmbedding(
35
+ model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
36
  )
37
  Settings.embed_model = embed_model
38
 
39
  #用初始化llm
40
  Settings.llm = llm
41
 
42
+ documents = SimpleDirectoryReader("data_domain").load_data()
43
  index = VectorStoreIndex.from_documents(documents)
44
  query_engine = index.as_query_engine()
45