Update main.py

main.py

@@ -35,7 +35,7 @@ def run():
                 )
                 with gr.Row():
                     btn_from_scratch = gr.Button("Start")
-                    btn_continue = gr.Button("Continue Generation")
+                    btn_continue = gr.Button("Generate Again")
                     btn_remove_last = gr.Button("Remove last instrument")
                     btn_regenerate_last = gr.Button("Regenerate last instrument")
             with gr.Column():
@@ -57,6 +57,18 @@ def run():
             empty_sequence = gr.Text(visible=False)
         with gr.Row():
             num_tokens = gr.Text(visible=False)
+        btn_from_scratch.click(
+            fn=generate_song,
+            inputs=[genre, temp, empty_sequence, qpm],
+            outputs=[
+                audio_output,
+                midi_file,
+                plot_output,
+                instruments_output,
+                text_sequence,
+                num_tokens,
+            ],
+        )            
         btn_continue.click(
             fn=generate_song,
             inputs=[genre, temp, text_sequence, qpm],