AsoulOfficial commited on
Commit
f4054f9
·
verified ·
1 Parent(s): e98c16c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
4
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
5
  from llama_index.legacy.callbacks import CallbackManager
6
  from llama_index.llms.openai_like import OpenAILike
 
7
 
8
  # Create an instance of CallbackManager
9
  callback_manager = CallbackManager()
@@ -27,7 +28,7 @@ st.title("llama_index")
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
 
 
4
  from llama_index.embeddings.huggingface import HuggingFaceEmbedding
5
  from llama_index.legacy.callbacks import CallbackManager
6
  from llama_index.llms.openai_like import OpenAILike
7
+ from sentence_transformers import SentenceTransformer # 确保你已经安装了 sentence-transformers 库
8
 
9
  # Create an instance of CallbackManager
10
  callback_manager = CallbackManager()
 
28
  @st.cache_resource
29
  def init_models():
30
  embed_model = HuggingFaceEmbedding(
31
+ model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2" # 这里替换成你想要使用的在线模型名称
32
  )
33
  Settings.embed_model = embed_model
34