modified: app.py

app.py

@@ -380,7 +380,7 @@ def generate_music(
 
 
 @spaces.GPU(duration=120)
-def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=200):
+def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=2):
 
     # Ensure the output folder exists
     output_dir = "./output"
@@ -395,7 +395,7 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=
         
         return music
     except Exception as e:
-        gr.Warning("An Error Occured: " + e)
+        gr.Warning("An Error Occured: " + str(e))
         return none
     finally:
         print("Temporary files deleted.")
@@ -425,7 +425,7 @@ with gr.Blocks() as demo:
                 
             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=100, maximum=3000, step=100, value=500, interactive=True)
+                max_new_tokens = gr.Slider(label="Duration of song", minimum=1, maximum=30, step=1, value=5, interactive=True)
                 submit_btn = gr.Button("Submit")
                 music_out = gr.Audio(label="Audio Result")