Update app.py

app.py

@@ -13,6 +13,7 @@ import numpy as np
 from huggingface_hub import hf_hub_download, list_repo_files
 import boto3
 from botocore.exceptions import NoCredentialsError
+from hashlib import md5
 
 # Define constants
 DESCRIPTION = "[ShowUI Demo](https://huggingface.co/showlab/ShowUI-2B)"
@@ -222,14 +223,18 @@ def build_demo(embed_mode, concurrency_count=1):
                     ["./examples/safari_google.png", "Click on search bar."],
                 ]
 
+                def compute_image_hash(image_array):
+                    """Compute a hash for the image."""
+                    img = Image.fromarray(np.uint8(image_array))
+                    return md5(img.tobytes()).hexdigest()
+
                 def set_example_flag(image_array, query):
-                    image = Image.fromarray(np.uint8(image_array))
+                    image_hash = compute_image_hash(image_array)
                     for example_path, _ in examples:
                         example_image = Image.open(example_path)
-                        if np.array_equal(np.array(image), np.array(example_image)):
-                            print("same")
+                        example_hash = md5(example_image.tobytes()).hexdigest()
+                        if image_hash == example_hash:
                             return True
-                    print("different")
                     return False
 
                 gr.Examples(