Update app.py

app.py

@@ -257,6 +257,13 @@ def preview_masked_area(audio_path, mask_start_point, mask_end_point):
     color_masked_spec_image.save(masked_spec_image_path)
 
     return masked_spec_image_path
+
+def load_inpaint_example(prompt_inp, audio_path):
+
+    in_spec_path = load_input_spectrogram(audio_path)
+    masked_spec_path = preview_masked_area(audio_path, 256, 768)
+    
+    return in_spec_path, masked_spec_path
     
 css="""
 div#col-container{
@@ -316,13 +323,20 @@ with gr.Blocks(css=css) as demo:
                     input_spectrogram = gr.Image(label="Input Spectrogram")
                     output_spectrogram = gr.Image(label="Output Spectrogram")
     
-            
+            gr.Examples(
+                examples = [
+                    ["Ambulance siren", "./notebooks/examples/img2img/GIOApFAWDOc_160.wav"],
+                    ["A cat is moewing", "./notebooks/examples/img2img/YniwgMbB6tpQ_01.wav"],
+                    ["A car racing", "./notebooks/examples/img2img/_GI7meqlYZk_30.wav"]
+                ],
+                inputs = [prompt_img2img, audio_in_img2img]
+            )
 
             submit_btn_img2img.click(
                 fn = infer_img2img,
                 inputs = [prompt_img2img, audio_in_img2img, prompt_strength],
                 outputs = [audio_out_img2img, input_spectrogram, output_spectrogram]
-            )
+            )     
 
         with gr.Tab("Audio InPainting"):
             prompt_inp = gr.Textbox(label="Prompt")
@@ -343,6 +357,19 @@ with gr.Blocks(css=css) as demo:
                     input_spectrogram_inp = gr.Image(label="Input Spectrogram")
                     output_spectrogram_inp = gr.Image(label="Output Spectrogram")
 
+            gr.Examples(
+                examples = [
+                    ["A siren ringing with a vehicle speeding closer", "./notebooks/examples/inpainting/IvfaKPDWC00_160.wav"],
+                    ["A woman speaking", "./notebooks/examples/inpainting/9z8XIRyUq9Q_30.wav"],
+                    ["An infant crying", "./notebooks/examples/inpainting/14ekd4nkpwc_28.wav"],
+                    ["A dog barking and growling", "./notebooks/examples/inpainting/3ek-xLwr05Q_30.wav"]
+                ],
+                fn = load_inpaint_example,
+                inputs = [prompt_inp, audio_in_inp],
+                outputs = [masked_spec_preview],
+                queue = False
+            )
+
             audio_in_inp.upload(
                 fn = load_input_spectrogram,
                 inputs = [audio_in_inp],