tushifire commited on
Commit
1aac8ff
·
1 Parent(s): 9ccddbe

st.cache_resource may need streamlit update from hf

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,7 +10,7 @@ paper_you_like = st.text_input(
10
  placeholder = None
11
  )
12
 
13
- @st.cache_resource
14
  def get_sentences_data():
15
  with open('sentences.pkl', 'rb') as f:
16
  sentences = pickle.load(f)
@@ -19,7 +19,7 @@ def get_sentences_data():
19
  sentences = get_sentences_data()
20
 
21
 
22
- @st.cache_resource
23
  def get_embeddings_data():
24
  with open('embeddings.pkl', 'rb') as f:
25
  embeddings = pickle.load(f)
@@ -28,7 +28,7 @@ def get_embeddings_data():
28
  embeddings = get_embeddings_data()
29
 
30
 
31
- @st.cache_resource
32
  def get_model():
33
  model = SentenceTransformer('all-MiniLM-L6-v2')
34
  return model
 
10
  placeholder = None
11
  )
12
 
13
+ # @st.cache_resource
14
  def get_sentences_data():
15
  with open('sentences.pkl', 'rb') as f:
16
  sentences = pickle.load(f)
 
19
  sentences = get_sentences_data()
20
 
21
 
22
+ # @st.cache_resource
23
  def get_embeddings_data():
24
  with open('embeddings.pkl', 'rb') as f:
25
  embeddings = pickle.load(f)
 
28
  embeddings = get_embeddings_data()
29
 
30
 
31
+ # @st.cache_resource
32
  def get_model():
33
  model = SentenceTransformer('all-MiniLM-L6-v2')
34
  return model