Update app.py

app.py

@@ -240,7 +240,7 @@ def generate_music(
                     use_cache=True,
                     top_k=50,
                     num_beams=1,
-                    # assistant_model=assistant_model
+                    assistant_model=assistant_model
                 )
                 if output_seq[0][-1].item() != mmtokenizer.eoa:
                     tensor_eoa = torch.as_tensor([[mmtokenizer.eoa]]).to(model.device)
@@ -427,26 +427,26 @@ with gr.Blocks() as demo:
                 submit_btn = gr.Button("Submit")
                 music_out = gr.Audio(label="Audio Result")
 
-        gr.Examples(
-            examples=[
-                ["Rap, Hip-Hop, Street Vibes, Tough, Piercing Vocals, Piano, Synthesizer, Clear Male Vocals",
-                """[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
-                """],
-            ],
-            inputs=[genre_txt, lyrics_txt],
-            outputs=[music_out],
-            cache_examples=True,
-            cache_mode="eager",
-            fn=infer
-        )
+#         gr.Examples(
+#             examples=[
+#                 ["Rap, Hip-Hop, Street Vibes, Tough, Piercing Vocals, Piano, Synthesizer, Clear Male Vocals",
+#                 """[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
+#                 """],
+#             ],
+#             inputs=[genre_txt, lyrics_txt],
+#             outputs=[music_out],
+#             cache_examples=True,
+#             cache_mode="eager",
+#             fn=infer
+#         )
         
         gr.Examples(
             examples=[
@@ -488,7 +488,7 @@ Living out my dreams with this mic and a deal
             inputs=[genre_txt, lyrics_txt],
             outputs=[music_out],
             cache_examples=True,
-            cache_mode="lazy",
+            cache_mode="eager",
             fn=infer
         )