searches execute

1 files changed, 5 insertions, 4 deletions

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index 36563910..2f78ecd3 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -70,7 +70,7 @@ def upload(name):
   query = np.array([ vec ]).astype('float32')
   
   # query FAISS!
-  distances, indexes = faiss_dataset.search(query, 5)
+  distances, indexes = faiss_dataset.search(query, 10)
 
   if len(indexes) == 0:
     print("weird, no results!")
@@ -85,6 +85,7 @@ def upload(name):
     return []
 
   lookup = {}
+  ids = [i+1 for i in indexes]
   for _d, _i in zip(distances, indexes):
     lookup[_i+1] = _d
 
@@ -97,13 +98,13 @@ def upload(name):
   query = {
     'timing': time.time() - start,
   }
-  results = [ dataset.get_identity(index) for index in indexes ]
+  results = [ dataset.get_identity(id) for id in ids ]
 
   print(results)
   return jsonify({
     'results': results,
-    # 'distances': distances.tolist(),
-    # 'indexes': indexes.tolist(),
+    'distances': distances.tolist(),
+    'indexes': indexes.tolist(),
   })
 
 @api.route('/dataset/<name>/name', methods=['GET'])