Spaces:
Runtime error
Runtime error
Commit
·
22fe311
1
Parent(s):
a70ef44
Update image_search.py
Browse files- 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",
|