diff --git a/environment.yml b/environment.yml index e56ebe26..e2e3245d 100644 --- a/environment.yml +++ b/environment.yml @@ -6,7 +6,7 @@ dependencies: - numpy=1.* - pytorch=1.* - transformers=4.* - - sentence-transformers=2.0.0 + - sentence-transformers=2.1.0 - fastapi=0.* - uvicorn=0.* - pyyaml=5.* diff --git a/src/search_type/image_search.py b/src/search_type/image_search.py index 34a38a9e..f0025328 100644 --- a/src/search_type/image_search.py +++ b/src/search_type/image_search.py @@ -160,7 +160,7 @@ def setup(config: ImageSearchConfig, regenerate: bool, verbose: bool) -> ImageSe # Extract Entries image_directory = resolve_absolute_path(config.input_directory, strict=True) - image_names = extract_entries(image_directory, config.verbose) + image_names = extract_entries(image_directory, verbose) # Compute or Load Embeddings embeddings_file = resolve_absolute_path(config.embeddings_file)