summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan/extract_dense_vectors.py
diff options
context:
space:
mode:
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
-rw-r--r--cli/app/commands/biggan/extract_dense_vectors.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py
index 1f7f8c3..fca66bd 100644
--- a/cli/app/commands/biggan/extract_dense_vectors.py
+++ b/cli/app/commands/biggan/extract_dense_vectors.py
@@ -1,7 +1,7 @@
import click
import os
-from app.utils.cortex_utils import fetch_cortex_folder, find_unprocessed_files
+from app.utils.cortex_utils import download_cortex_files, find_unprocessed_files
from app.search.search_class import find_nearest_vector_for_images
from app.search.search_dense import find_dense_embedding_for_images
from app.search.json import params_dense_dict