Update app.py

app.py

@@ -121,17 +121,8 @@ with gr.Blocks(css=css) as demo:
         ["examples/3.png", 1, "A crowded city", 500]
     ]
 
-    def on_example_click(*args, **kwargs) -> Tuple[str, str]:
-        example_image = args[0]
-        
-        # Map the image file to the correct example index (1, 2, 3)
-        example_map = {
-            "examples/1.png": 1,
-            "examples/2.png": 2,
-            "examples/3.png": 3
-        }
-        
-        example_index = example_map.get(example_image, 1)  # Default to 1 if not found
+    def update_examples(index):
+        example_index = int(index)  # Convert index to integer for use
         return load_cached_example_outputs(example_index)
 
     gr.Examples(