updating face search API

1 files changed, 2 insertions, 5 deletions

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index 743e06f4..af3db4d0 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -113,8 +113,7 @@ def upload(dataset_name):
       dists.append(round(float(_d), 2))
       ids.append(_i+1)
 
-  file_records = [ dataset.get_file_record(int(_i)) for _i in ids ]
-  identities = [ dataset.get_identity(rec.identity_id) for rec in file_records ]
+  identities = [ dataset.get_identity(int(_i)) for _i in ids ]
 
   # print(distances)
   # print(ids)
@@ -151,8 +150,6 @@ def name_lookup(dataset_name):
     'timing': time.time() - start,
   }
 
-  print(terms)
-
   if len(terms) == 0:
     return jsonify({ 'query': query, 'results': [] })
 
@@ -193,7 +190,7 @@ def name_lookup(dataset_name):
   sorted_names = sorted(lookup.items(), key=operator.itemgetter(1), reverse=True)[0:10]
   top_names = [results_lookup[item[0]] for item in sorted_names]
   results = dataset.get_file_records_for_identities(top_names)
-  
+
   print(results)
   return jsonify({
     'query': query,