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)