Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def calculate_similarity(img1, img2):
|
|
22 |
def image_similarity(target_image, image_list):
|
23 |
target_image = target_image.astype(np.uint8)
|
24 |
image_list = [image.astype(np.uint8) for image in image_list]
|
25 |
-
scores =
|
26 |
results = list(zip(image_list, scores))
|
27 |
results.sort(key=lambda x: x[1], reverse=True) # Sort by similarity score
|
28 |
return [str(result) for result in results]
|
|
|
22 |
def image_similarity(target_image, image_list):
|
23 |
target_image = target_image.astype(np.uint8)
|
24 |
image_list = [image.astype(np.uint8) for image in image_list]
|
25 |
+
scores = calculate_similarity(target_image, image_list)
|
26 |
results = list(zip(image_list, scores))
|
27 |
results.sort(key=lambda x: x[1], reverse=True) # Sort by similarity score
|
28 |
return [str(result) for result in results]
|