fix

2 files changed, 9 insertions, 5 deletions

diff --git a/cli/app/commands/cortex/upload.py b/cli/app/commands/cortex/upload.py
index c1b0a46..c176eb9 100644
--- a/cli/app/commands/cortex/upload.py
+++ b/cli/app/commands/cortex/upload.py
@@ -1,6 +1,7 @@
 import click
+import json
 
-from app.utils.cortex_utils import upload_to_cortex
+from app.utils.cortex_utils import upload_file_to_cortex
 
 @click.command('')
 @click.option('-i', '--input', 'opt_fp_in', required=True, 
@@ -12,4 +13,5 @@ def cli(ctx, opt_fp_in, opt_folder_id):
   """
   Test uploading a file to Cortex
   """
-  upload_to_cortex(opt_folder_id, opt_fp_in)
+  data = upload_file_to_cortex(opt_folder_id, opt_fp_in)
+  print(json.dumps(data, indent=2))
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index e006fdd..c7cf078 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -4,7 +4,8 @@ import itertools
 import numpy as np
 from io import BytesIO
 import os
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 
+import json
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
 from PIL import Image
 import scipy
@@ -423,8 +424,9 @@ def find_dense_embedding_for_images(params):
       fp = BytesIO()
       image.save(fp, format='png')
       data = upload_bytes_to_cortex(params.folder_id, sample_fn + "-inverse.png", fp, "image/png")
-      if data is not None:
-        file_id = data['id']
+      print(json.dumps(data, indent=2))
+      if data is not None and 'files' in data:
+        file_id = data['files'][0]['id']
         fp_out_pkl = os.path.join(app_cfg.DIR_VECTORS, "file_{}.pkl".format(file_id))
         out_data = {
           'id': file_id,