Hugging Face's logo

Spaces:
heikowagner
/
GPT-Docker
Build error

App Files Community
4
heikowagner commited on
Commit
9457dc8
•
1 Parent(s): aeb550e

emb

Browse files
Files changed (1) hide show
  1. app/load_model.py +1 -1
app/load_model.py CHANGED
@@ -88,7 +88,7 @@ def load_openai_model():
88
  def load_openai_embedding():
89
  return OpenAIEmbeddings()
90
 
91
- #@st.cache_resource
92
  def load_embedding(model_name):
93
  embeddings = HuggingFaceInstructEmbeddings(
94
  query_instruction="Represent the query for retrieval: ",
 
88
  def load_openai_embedding():
89
  return OpenAIEmbeddings()
90
 
91
+ @st.cache_resource
92
  def load_embedding(model_name):
93
  embeddings = HuggingFaceInstructEmbeddings(
94
  query_instruction="Represent the query for retrieval: ",