zabir-nabil commited on
Commit
22fe311
1 Parent(s): a70ef44

Update image_search.py

Browse files
Files changed (1) hide show
  1. image_search.py +2 -2
image_search.py CHANGED
@@ -75,9 +75,9 @@ def search_images(search_text, image_path, k = 1):
75
  for i in range(len(top_k_indices)):
76
  print(f"{image_filenames[int(top_k_indices[i])]} :: {top_k_vals[i]}")
77
  images_ret.append(image_filenames[int(top_k_indices[i])])
78
- scores_ret.append(top_k_vals[i])
79
 
80
- return top_k_vals, top_k_indices
81
 
82
  if __name__ == '__main__':
83
  img_filenames = ["demo_images/1280px-Cox's_Bazar_Sunset.JPG", "demo_images/Cox's_Bazar,_BangladeshThe_sea_is_calm.jpg", "demo_images/Panta_Vaat_Hilsha_Fisha_VariousVarta_2012.JPG",
 
75
  for i in range(len(top_k_indices)):
76
  print(f"{image_filenames[int(top_k_indices[i])]} :: {top_k_vals[i]}")
77
  images_ret.append(image_filenames[int(top_k_indices[i])])
78
+ scores_ret.append(float(top_k_vals[i]))
79
 
80
+ return images_ret, scores_ret, top_k_vals, top_k_indices
81
 
82
  if __name__ == '__main__':
83
  img_filenames = ["demo_images/1280px-Cox's_Bazar_Sunset.JPG", "demo_images/Cox's_Bazar,_BangladeshThe_sea_is_calm.jpg", "demo_images/Panta_Vaat_Hilsha_Fisha_VariousVarta_2012.JPG",