Update app.py

app.py

@@ -124,7 +124,7 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
         "--genre_txt", f"{genre_txt_path}",
         "--lyrics_txt", f"{lyrics_txt_path}",
         "--run_n_segments", f"{num_segments}",
-        "--stage2_batch_size", "2",
+        "--stage2_batch_size", "4",
         "--output_dir", f"{output_dir}",
         "--cuda_idx", "0",
         "--max_new_tokens", f"{max_new_tokens}",
@@ -193,11 +193,22 @@ with gr.Blocks() as demo:
             with gr.Column():
                 genre_txt = gr.Textbox(label="Genre")
                 lyrics_txt = gr.Textbox(label="Lyrics")
-                gr.Examples(
-                    examples = [
-                        [
-                            "female blues airy vocal bright vocal piano sad romantic guitar jazz",
-                            """[verse]
+                
+            with gr.Column():
+                if is_shared_ui:
+                    num_segments = gr.Number(label="Number of Song Segments", value=2, interactive=False)
+                    max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum="3000", step=500, value=1500, interactive=False)
+                else:
+                    num_segments = gr.Number(label="Number of Song Segments", value=2, interactive=True)
+                    max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum="24000", step=500, value=3000, interactive=True)
+                submit_btn = gr.Button("Submit")
+                music_out = gr.Audio(label="Audio Result")
+
+        gr.Examples(
+            examples = [
+                [
+                    "female blues airy vocal bright vocal piano sad romantic guitar jazz",
+                    """[verse]
 In the quiet of the evening, shadows start to fall
 Whispers of the night wind echo through the hall
 Lost within the silence, I hear your gentle voice
@@ -208,20 +219,30 @@ Don't let this moment fade, hold me close tonight
 With you here beside me, everything's alright
 Can't imagine life alone, don't want to let you go
 Stay with me forever, let our love just flow
-                            """
-                        ]
-                    ], 
-                    inputs = [genre_txt, lyrics_txt]
-                )
-            with gr.Column():
-                if is_shared_ui:
-                    num_segments = gr.Number(label="Number of Song Segments", value=2, interactive=False)
-                    max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum="24000", step=500, value=3000, interactive=False)
-                else:
-                    num_segments = gr.Number(label="Number of Song Segments", value=2, interactive=True)
-                    max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum="24000", step=500, value=3000, interactive=True)
-                submit_btn = gr.Button("Submit")
-                music_out = gr.Audio(label="Audio Result")
+                    """
+                ],
+                [
+                    "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
+City lights are fading, but my vision's clear
+Got my team beside me, no room for fear
+Walking through the streets, beats inside my head
+Every step I take, closer to the bread
+People passing by, they don't understand
+Building up my future with my own two hands
+
+[chorus]
+This is my life, and I'm aiming for the top
+Never gonna quit, no, I'm never gonna stop
+Through the highs and lows, I'mma keep it real
+Living out my dreams with this mic and a deal
+                    """
+                ]
+            ], 
+             inputs = [genre_txt, lyrics_txt]
+        )
     
     submit_btn.click(
         fn = infer,