1 files changed, 15 insertions, 0 deletions
diff --git a/cli/app/commands/biggan/search_dense.py b/cli/app/commands/biggan/search_dense.py
new file mode 100644
index 0000000..0cf56a3
--- /dev/null
+++ b/cli/app/commands/biggan/search_dense.py
@@ -0,0 +1,15 @@
+import click
+
+from app.search.params import Params
+from app.search.search_dense import find_dense_embedding_for_images
+
+@click.command('')
+@click.option('-i', '--input', 'opt_fp_in', required=True, 
+  help='Path to input image')
+@click.pass_context
+def cli(ctx, opt_fp_in):
+  """
+  Search for an image (class vector) in BigGAN using gradient descent
+  """
+  params = Params(opt_fp_in)
+  find_dense_embedding_for_images(params)