mpsk commited on
Commit
0c1a701
1 Parent(s): f1a5f04

fixed Top-K Images returns nothing

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -307,7 +307,7 @@ if 'xq' in st.session_state:
307
  with st.expander("Top-K Images"):
308
  with st.container():
309
  boxes_w_img, _ = postprocess(o_matches, st.session_state.text_prompts,
310
- None)
311
  boxes_w_img = sorted(
312
  boxes_w_img, key=lambda x: x[4], reverse=True)
313
  for img_id, img_url, img_w, img_h, img_score, boxes in boxes_w_img:
 
307
  with st.expander("Top-K Images"):
308
  with st.container():
309
  boxes_w_img, _ = postprocess(o_matches, st.session_state.text_prompts,
310
+ o_matches)
311
  boxes_w_img = sorted(
312
  boxes_w_img, key=lambda x: x[4], reverse=True)
313
  for img_id, img_url, img_w, img_h, img_score, boxes in boxes_w_img: