klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 13:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Image input directory not required by collate result as image_name already absolute path

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-20 02:56:23 +04:00
parent 6c9ffdba57
commit 2a5445216c
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/search_type/image_search.py
View File

@@ -169,13 +169,11 @@ def render_results(hits, image_names, image_directory, count):
img.show()
def collate_results(hits, image_names, image_directory, output_directory, static_files_url, count=5):
def collate_results(hits, image_names, output_directory, static_files_url, count=5):
results = []
image_directory = resolve_absolute_path(image_directory, strict=True)
for index, hit in enumerate(hits[:count]):
source_image_name = image_names[hit['corpus_id']]
source_path = image_directory.joinpath(source_image_name)
source_path = image_names[hit['corpus_id']]
target_image_name = f"{index}{source_path.suffix}"
target_path = resolve_absolute_path(f"{output_directory}/{target_image_name}")