Update app.py

app.py

@@ -28,9 +28,17 @@ def image_similarity(target_image, image):
     target_image = target_image.astype(np.uint8)
     image = image.astype(np.uint8)
     scores = compute_similarity(target_image, [image])
-    result = (str(image.tolist()), f"Score: {scores[0]:.4f}")
+    result = (format_image(image), f"Score: {scores[0]:.4f}")
     return [result]
 
+# Function to format the image representation as a string
+def format_image(image):
+    formatted_image = ""
+    for row in image:
+        formatted_row = ", ".join([f"[{', '.join(map(str, pixel))}]" for pixel in row])
+        formatted_image += f"[{formatted_row}], "
+    return formatted_image[:-2]
+
 # Prepare Gradio interface
 iface = gr.Interface(
     fn=image_similarity,