Update app.py

app.py

@@ -24,26 +24,31 @@ def compute_similarity(target_image, image_list):
 
 
 # Function to handle the Gradio interface
-def image_similarity(target_image, image_list):
+def image_similarity(target_image, image):
     target_image = target_image.astype(np.uint8)
-    image_list = [image.astype(np.uint8) for image in image_list]
-    scores = compute_similarity(target_image, image_list)
-    results = list(zip(image_list, scores))
-    results.sort(key=lambda x: x[1], reverse=True)  # Sort by similarity score
-    formatted_results = [(str(image.tolist()), f"Score: {score:.4f}") for image, score in results]
-    return formatted_results
+    image = image.astype(np.uint8)
+    scores = compute_similarity(target_image, [image])
+    result = (str(image.tolist()), f"Score: {scores[0]:.4f}")
+    return [result]
 
 # Prepare Gradio interface
 iface = gr.Interface(
     fn=image_similarity,
     inputs=[
         gr.inputs.Image(type="numpy", label="Target Image"),
-        gr.inputs.Image(type="numpy", label="Image List")
+        gr.inputs.Image(type="numpy", label="Image")
     ],
     outputs="text",
     title="Image Similarity Calculator",
-    description="Upload an image and a list of images. Get similarity scores."
+    description="Upload a target image and another image. Get the similarity score."
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()
+
+
+
+
+
+
+