Update app.py

app.py

@@ -25,6 +25,13 @@ model.setup()
 
 CACHE_DIR = "gradio_cached_examples"
 
+
+example_mapping = {
+    1: os.path.join(CACHE_DIR, "1"),
+    2: os.path.join(CACHE_DIR, "2"),
+    3: os.path.join(CACHE_DIR, "3")
+}
+
 #for local cache
 def load_cached_example_outputs(example_index: int) -> Tuple[str, str]:
     cached_dir = os.path.join(CACHE_DIR, str(example_index))  # Use the example index to find the directory
@@ -126,16 +133,11 @@ with gr.Blocks(css=css) as demo:
         return load_cached_example_outputs(example_index)
 
     gr.Examples(
-        examples=[
-            ["examples/1.png", 3, "A scenic mountain view", 500],  # Example 1
-            ["examples/2.png", 2, "A forest with birds", 500],    # Example 2
-            ["examples/3.png", 1, "A crowded city", 500]          # Example 3
-        ],
+        examples=examples,  # Example inputs
         inputs=[image, num_audios, prompt, steps],
         outputs=[processed_image, generated_audio],
         cache_examples=True,  # Cache examples to avoid running the model
-        fn=update_examples,   # Call function that loads example based on index
-        inputs=[index_selector]  # Use the dropdown to select the example index
+        fn=lambda *args: on_example_click(int(args[0].split('/')[-1][0]))  # Extract example index from image path
     )
 
     gr.on(