Ceyda Cinarel commited on
Commit
7dbbdfd
1 Parent(s): 6d30f20

add calculating spinner

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -117,16 +117,20 @@ query_images = q_cols[2].file_uploader("(optional) Upload images to rank them",t
117
 
118
  if query_images:
119
  st.write("Ranking your uploaded images with respect to input text:")
120
- ids, dists = query_with_images(query_images,query_text)
 
121
  else:
122
  st.write("Found these images within validation set:")
123
- proc = processor(text=[query_text], images=None, return_tensors="jax", padding=True)
124
- vec = np.asarray(model.get_text_features(**proc))
125
- ids, dists = image_index.knnQuery(vec, k=top_k)
 
126
 
127
  show_gif=st.checkbox("Play GIFs",value=True,help="Will play the original animation. Only first frame is used in training!")
128
  ext = "jpg" if not show_gif else "gif"
129
  res_cols=st.beta_columns(col_count)
 
 
130
  for i,(id_, dist) in enumerate(zip(ids, dists)):
131
  j=i%col_count
132
  with res_cols[j]:
 
117
 
118
  if query_images:
119
  st.write("Ranking your uploaded images with respect to input text:")
120
+ with st.spinner("Calculating..."):
121
+ ids, dists = query_with_images(query_images,query_text)
122
  else:
123
  st.write("Found these images within validation set:")
124
+ with st.spinner("Calculating..."):
125
+ proc = processor(text=[query_text], images=None, return_tensors="jax", padding=True)
126
+ vec = np.asarray(model.get_text_features(**proc))
127
+ ids, dists = image_index.knnQuery(vec, k=top_k)
128
 
129
  show_gif=st.checkbox("Play GIFs",value=True,help="Will play the original animation. Only first frame is used in training!")
130
  ext = "jpg" if not show_gif else "gif"
131
  res_cols=st.beta_columns(col_count)
132
+
133
+
134
  for i,(id_, dist) in enumerate(zip(ids, dists)):
135
  j=i%col_count
136
  with res_cols[j]: