Update app.py

app.py

@@ -2,6 +2,7 @@ import numpy as np
 from skimage.metrics import structural_similarity as ssim
 import gradio as gr
 import cv2
+import os
 
 # Function to calculate SSIM between two images
 def calculate_similarity(img1, img2):
@@ -25,7 +26,7 @@ def compute_similarity(target_image, image_list):
 # Function to handle the Gradio interface
 def image_similarity(target_image, image_folder):
     target_image = target_image.astype(np.uint8)
-    image_paths = gr.Interface.get_filepaths(image_folder)
+    image_paths = [os.path.join(image_folder, file) for file in os.listdir(image_folder) if os.path.isfile(os.path.join(image_folder, file))]
     scores = compute_similarity(target_image, image_paths)
     results = []
     for image_path, score in zip(image_paths, scores):
@@ -38,7 +39,7 @@ iface = gr.Interface(
     fn=image_similarity,
     inputs=[
         gr.inputs.Image(type="numpy", label="Target Image"),
-        gr.inputs.ImageFolder(label="Image Folder")
+        gr.inputs.Folder(label="Image Folder")
     ],
     outputs="text",
     title="Image Similarity Calculator",
@@ -47,4 +48,3 @@ iface = gr.Interface(
 
 # Launch the interface
 iface.launch()
-