revised query strategy

app.py

@@ -118,7 +118,7 @@ def query(xq, exclude_list=None):
                 IMG_DB_NAME,
                 OBJ_DB_NAME,
                 exclude_list=exclude_list,
-                topk=5000,
+                topk=10,
             )
             img_ids = [r["img_id"] for r in matches]
             if "topk_img_id" not in st.session_state:
@@ -141,7 +141,7 @@ def query(xq, exclude_list=None):
                 IMG_DB_NAME,
                 OBJ_DB_NAME,
                 thresh=-1,
-                topk=5000,
+                topk=10,
             )
             status_bar[0].write("Retrieving Non-TopK in Another TopK Images...")
             pbar.progress(75)

query_model.py

@@ -22,8 +22,8 @@ def topk_obj_query(client, xq, IMG_DB_NAME, OBJ_DB_NAME,
             WHERE obj_id IN (
                 SELECT obj_id FROM (
                     SELECT obj_id, distance(prelogit, {_xq}) AS dist FROM {OBJ_DB_NAME}
-                    ORDER BY dist DESC LIMIT {topk}
-                ) {_cond} LIMIT 10
+                    ORDER BY dist DESC LIMIT 5000
+                ) {_cond} LIMIT {topk}
             )
             """)
     _subq_str = ' UNION ALL '.join(_subq_str)