SiraH commited on
Commit
11debf7
1 Parent(s): 833996b

change model embddding

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -197,7 +197,7 @@ def set_custom_prompt():
197
 
198
  @st.cache_resource
199
  def load_embeddings():
200
- embeddings = HuggingFaceEmbeddings(model_name = "all-MiniLM-L6-v2",
201
  model_kwargs = {'device': 'cpu'})
202
  return embeddings
203
 
 
197
 
198
  @st.cache_resource
199
  def load_embeddings():
200
+ embeddings = HuggingFaceEmbeddings(model_name = "thenlper/gte-base",
201
  model_kwargs = {'device': 'cpu'})
202
  return embeddings
203