Update app.py

app.py

@@ -13,7 +13,6 @@ 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)"
@@ -212,36 +211,30 @@ def build_demo(embed_mode, concurrency_count=1):
                 submit_btn = gr.Button(value="Submit", variant="primary")
 
                 examples = [
-                    ["./examples/app_store.png", "Download Kindle."],
-                    ["./examples/ios_setting.png", "Turn off Do not disturb."],
-                    ["./examples/apple_music.png", "Star to favorite."],
-                    ["./examples/map.png", "Boston."],
-                    ["./examples/wallet.png", "Scan a QR code."],
-                    ["./examples/word.png", "More shapes."],
-                    ["./examples/web_shopping.png", "Proceed to checkout."],
-                    ["./examples/web_forum.png", "Post my comment."],
-                    ["./examples/safari_google.png", "Click on search bar."],
+                    ["./examples/app_store.png", "Download Kindle.", True],
+                    ["./examples/ios_setting.png", "Turn off Do not disturb.", True],
+                    ["./examples/apple_music.png", "Star to favorite.", True],
+                    ["./examples/map.png", "Boston.", True],
+                    ["./examples/wallet.png", "Scan a QR code.", True],
+                    ["./examples/word.png", "More shapes.", True],
+                    ["./examples/web_shopping.png", "Proceed to checkout.", True],
+                    ["./examples/web_forum.png", "Post my comment.", True],
+                    ["./examples/safari_google.png", "Click on search bar.", True],
                 ]
 
-                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_hash = compute_image_hash(image_array)
-                    for example_path, _ in examples:
+                def check_if_example(image_array, query):
+                    """Check if the image and query match an example."""
+                    for example_path, example_query, is_example in examples:
                         example_image = Image.open(example_path)
-                        example_hash = md5(example_image.tobytes()).hexdigest()
-                        if image_hash == example_hash:
-                            return True
+                        if np.array_equal(np.array(example_image), image_array) and example_query == query:
+                            return is_example
                     return False
 
                 gr.Examples(
-                    examples=examples,
+                    examples=[[e[0], e[1]] for e in examples],
                     inputs=[imagebox, textbox],
                     outputs=[state_is_example_image],
-                    fn=set_example_flag,
+                    fn=check_if_example,
                     examples_per_page=3
                 )