Update app.py

app.py

@@ -10,12 +10,13 @@ def calculate_similarity(img1, img2):
     return ssim(img1, img2)
 
 # Function to compute similarity scores for all images
-def compute_similarity(target_image, image_list):
-    scores = []
-    for image in image_list:
-        similarity_score = calculate_similarity(target_image, image)
-        scores.append(similarity_score)
-    return scores
+def calculate_similarity(img1, img2):
+    if len(img1.shape) == 2:
+        img1 = cv2.cvtColor(img1, cv2.COLOR_GRAY2RGB)
+    if len(img2.shape) == 2:
+        img2 = cv2.cvtColor(img2, cv2.COLOR_GRAY2RGB)
+    return ssim(img1, img2)
+
 
 # Function to handle the Gradio interface
 def image_similarity(target_image, image_list):