Update app.py

app.py

@@ -222,10 +222,10 @@ def build_demo(embed_mode, concurrency_count=1):
                     ["./examples/safari_google.png", "Click on search bar.", True],
                 ]
 
-                def check_if_example(image_path, query):
-                    """Check if the image and query match an example using the flag."""
-                    for example_path, example_query, is_example in examples:
-                        if example_path == image_path and example_query == query:
+                def check_if_example(query):
+                    """Check if the query matches an example."""
+                    for _, example_query, is_example in examples:
+                        if example_query == query:
                             return is_example
                     return False
 
@@ -233,7 +233,7 @@ def build_demo(embed_mode, concurrency_count=1):
                     examples=[[e[0], e[1]] for e in examples],
                     inputs=[imagebox, textbox],
                     outputs=[state_is_example_image],
-                    fn=check_if_example,
+                    fn=lambda img, query: check_if_example(query),
                     examples_per_page=3
                 )