JLD commited on
Commit
eef445c
1 Parent(s): f2611db

Trying to add embeddings

Browse files
Files changed (2) hide show
  1. .gitattributes +1 -0
  2. app.py +2 -2
.gitattributes CHANGED
@@ -25,3 +25,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ unsplash-25k-photos-embeddings.pkl filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -75,7 +75,7 @@ def load_model(name):
75
  # st.sidebar.success(f"Model {name} loaded")
76
  return model
77
 
78
- @st.cache
79
  def load_embeddings(filename):
80
  st.sidebar.info("Loading Unsplash-Lite image embeddings")
81
  with open(filename, "rb") as fIn:
@@ -83,7 +83,7 @@ def load_embeddings(filename):
83
  st.sidebar.success("Images embeddings loaded")
84
  return img_names, img_emb
85
 
86
- @st.cache
87
  def load_image_url_list(filename):
88
  url_list = pd.read_csv(filename, sep='\t', header=0)
89
  return url_list
 
75
  # st.sidebar.success(f"Model {name} loaded")
76
  return model
77
 
78
+ @st.cache(suppress_st_warning=True)
79
  def load_embeddings(filename):
80
  st.sidebar.info("Loading Unsplash-Lite image embeddings")
81
  with open(filename, "rb") as fIn:
 
83
  st.sidebar.success("Images embeddings loaded")
84
  return img_names, img_emb
85
 
86
+ @st.cache(suppress_st_warning=True)
87
  def load_image_url_list(filename):
88
  url_list = pd.read_csv(filename, sep='\t', header=0)
89
  return url_list