Update app.py

app.py

@@ -222,11 +222,13 @@ def build_demo(embed_mode, concurrency_count=1):
                     ["./examples/safari_google.png", "Click on search bar."],
                 ]
 
-                def set_example_flag(image, query):
-                    print(image)
-                    image_path = image if isinstance(image, str) else None
-                    is_example = any(image_path == example[0] for example in examples)
-                    return is_example
+                def set_example_flag(image_array, query):
+                    image = Image.fromarray(np.uint8(image_array))
+                    for example_path, _ in examples:
+                        example_image = Image.open(example_path)
+                        if np.array_equal(np.array(image), np.array(example_image)):
+                            return True
+                    return False
 
                 gr.Examples(
                     examples=examples,
@@ -298,4 +300,4 @@ if __name__ == "__main__":
         server_port=7860,
         ssr_mode=False,
         debug=True,
-    )
+    )