Update app.py

app.py

@@ -1,30 +1,23 @@
 import numpy as np
-from skimage.metrics import structural_similarity as ssim
 import gradio as gr
 import cv2
 
-# Function to calculate SSIM between two images
 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, win_size=3)
+    diff = cv2.absdiff(img1, img2)
+    similarity_score = np.mean(diff)
+    return similarity_score
 
-# Function to handle the Gradio interface
 def image_similarity(img1, img2):
     img1 = img1.astype(np.uint8)
     img2 = img2.astype(np.uint8)
-    if img1.shape != img2.shape:
-        max_height = max(img1.shape[0], img2.shape[0])
-        max_width = max(img1.shape[1], img2.shape[1])
-        img1 = cv2.resize(img1, (max_width, max_height))
-        img2 = cv2.resize(img2, (max_width, max_height))
     similarity_score = calculate_similarity(img1, img2)
     result = f"Similarity Score: {similarity_score:.4f}"
     return result
 
-# Prepare Gradio interface
 iface = gr.Interface(
     fn=image_similarity,
     inputs=[
@@ -36,5 +29,4 @@ iface = gr.Interface(
     description="Upload two images to compute their similarity."
 )
 
-# Launch the interface
 iface.launch()