util

1 files changed, 3 insertions, 1 deletions

diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index d41afc4..25d6ec7 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -21,6 +21,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
 sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), '../../../../live-cortex/rpc/'))
 from rpc import CortexRPC
 from app.search.params import timestamp
+from app.utils.cortex_utils import results_folder, upload_file_to_cortex
 from app.utils.tf_utils import read_checkpoint
 from subprocess import Popen, PIPE
 
@@ -442,7 +443,8 @@ class Listener:
     pipe.stdin.close()
     pipe.wait()
     print("Uploading video...")
-    data = upload_file_to_cortex(opt_folder_id, fp_out, datatype='video', activity='live')
+    folder = results_folder()
+    data = upload_file_to_cortex(folder['id'], fp_out, datatype='video', activity='live')
     print(json.dumps(data, indent=2))
     print("Done")