PrabakaranC commited on
Commit
ec0467a
1 Parent(s): d6147f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
5
  from flashrank import Ranker, RerankRequest
6
  import numpy as np
7
 
8
- @st.cache_data(allow_output_mutation=True)
9
  def get_embeddings():
10
  model_name = "BAAI/bge-large-en-v1.5"
11
  model_kwargs = {'device': 'cpu',"trust_remote_code":True}
 
5
  from flashrank import Ranker, RerankRequest
6
  import numpy as np
7
 
8
+ @st.cache_resource
9
  def get_embeddings():
10
  model_name = "BAAI/bge-large-en-v1.5"
11
  model_kwargs = {'device': 'cpu',"trust_remote_code":True}