diff options
| -rw-r--r-- | cli/app/search/live.py | 4 |
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") |