kaushalya commited on
Commit
58a320e
1 Parent(s): 9783cb8

Add larger embeddings file

Browse files
app.py CHANGED
@@ -14,7 +14,7 @@ def load_model():
14
 
15
  @st.cache(allow_output_mutation=True)
16
  def load_image_embeddings():
17
- embeddings_df = pd.read_pickle('feature_store/image_embeddings.pkl')
18
  image_embeds = np.stack(embeddings_df['image_embedding'])
19
  image_files = np.asarray(embeddings_df['files'].tolist())
20
  return image_files, image_embeds
 
14
 
15
  @st.cache(allow_output_mutation=True)
16
  def load_image_embeddings():
17
+ embeddings_df = pd.read_pickle('feature_store/image_embeddings_large.pkl')
18
  image_embeds = np.stack(embeddings_df['image_embedding'])
19
  image_files = np.asarray(embeddings_df['files'].tolist())
20
  return image_files, image_embeds
feature_store/image_embeddings_large.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dda0119181e05580b3080581f82ca0895b4bc785ae7e64c82c561397d2f761e8
3
+ size 14105206