Update app.py

app.py

@@ -123,7 +123,7 @@ def split_lyrics(lyrics: str):
 def generate_music(
         genre_txt=None,
         lyrics_txt=None,
-        run_n_segments=3,
+        run_n_segments=2,
         max_new_tokens=45,
         use_audio_prompt=False,
         audio_prompt_path="",
@@ -333,18 +333,17 @@ with gr.Blocks() as demo:
             with gr.Column():
                 genre_txt = gr.Textbox(label="Genre")
                 lyrics_txt = gr.Textbox(label="Lyrics")
-                use_audio_prompt = gr.Checkbox(label="Use Audio Prompt?", value=False)
+                use_audio_prompt = gr.Checkbox(label="Use Audio Prompt ?", value=False)
                 audio_prompt_input = gr.Audio(type="filepath", label="Audio Prompt (Optional)")
             with gr.Column():
                 num_segments = gr.Number(label="Number of Segments", value=2, interactive=True)
-                max_new_tokens = gr.Slider(label="Duration of song", minimum=1, maximum=50, step=1, value=30, interactive=True)
+                max_new_tokens = gr.Slider(label="Duration of song", minimum=1, maximum=45, step=1, value=30, interactive=True)
                 submit_btn = gr.Button("Submit")
                 music_out = gr.Audio(label="Mixed Audio Result")
                 with gr.Accordion(label="Vocal and Instrumental Result", open=False):
                     vocal_out = gr.Audio(label="Vocal Audio")
                     instrumental_out = gr.Audio(label="Instrumental Audio")
-        gr.Markdown("## Call for Contributions\nIf you find this space interesting please feel free to contribute.")
-
+                    
         # When the "Submit" button is clicked, pass the additional audio-related inputs to the function.
         submit_btn.click(
             fn=generate_music,
@@ -386,8 +385,18 @@ But they don't feel this love the way we do
 My heart beats only for you, can't you see?
 I won't let you slip away from me
                     """
-                ],
-["rap piano street tough piercing vocal hip-hop synthesizer clear vocal male",
+                ]
+            ],
+            inputs=[genre_txt, lyrics_txt],
+            outputs=[music_out, vocal_out, instrumental_out],
+            cache_examples=True,
+            cache_mode="eager",
+            fn=generate_music
+        )
+
+        gr.Examples(
+            examples=[ 
+                ["rap piano street tough piercing vocal hip-hop synthesizer clear vocal male",
                 """[verse]
 Woke up in the morning, sun is shining bright
 Chasing all my dreams, gotta get my mind right
@@ -433,8 +442,11 @@ Locked inside my mind, hot flame.
             inputs=[genre_txt, lyrics_txt],
             outputs=[music_out, vocal_out, instrumental_out],
             cache_examples=True,
-            cache_mode="eager",
+            cache_mode="lazy",
             fn=generate_music
         )
 
+gr.Markdown("## Call for Contributions\nIf you find this space interesting please 💖 this space and feel free to contribute.\n### TODO:\n1. [Aeromamba](https://huggingface.co/innova-ai/AEROMamba) for output enhancement.")
+
+
 demo.queue().launch(show_error=True)