Update app.py

app.py

@@ -8,7 +8,11 @@ def calculate_similarity(img1, img2):
         img1 = cv2.cvtColor(img1, cv2.COLOR_GRAY2RGB)
     if len(img2.shape) == 2:
         img2 = cv2.cvtColor(img2, cv2.COLOR_GRAY2RGB)
-    similarity_score = ssim(img1, img2, multichannel=True)
+    max_height = max(img1.shape[0], img2.shape[0])
+    max_width = max(img1.shape[1], img2.shape[1])
+    img1_resized = cv2.resize(img1, (max_width, max_height))
+    img2_resized = cv2.resize(img2, (max_width, max_height))
+    similarity_score = ssim(img1_resized, img2_resized, multichannel=True)
     return similarity_score
 
 def image_similarity(img1, img2):