no more duration params

app.py

@@ -18,18 +18,18 @@ def preprocess_audio(waveform):
     waveform_np = waveform.cpu().squeeze().numpy()
     return torch.from_numpy(waveform_np).unsqueeze(0).to(device)
 
-@spaces.GPU(duration=10)
-def generate_drum_sample() -> str:
-    model = MusicGen.get_pretrained('pharoAIsanders420/micro-musicgen-jungle')
-    model.set_generation_params(duration=10)
-    wav = model.generate_unconditional(1).squeeze(0)
+# @spaces.GPU(duration=10)
+# def generate_drum_sample() -> str:
+#     model = MusicGen.get_pretrained('pharoAIsanders420/micro-musicgen-jungle')
+#     model.set_generation_params(duration=10)
+#     wav = model.generate_unconditional(1).squeeze(0)
     
-    filename_without_extension = f'jungle'
-    filename_with_extension = f'{filename_without_extension}.wav'
+#     filename_without_extension = f'jungle'
+#     filename_with_extension = f'{filename_without_extension}.wav'
 
-    audio_write(filename_without_extension, wav.cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
+#     audio_write(filename_without_extension, wav.cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
 
-    return filename_with_extension
+#     return filename_with_extension
 
 @spaces.GPU(duration=10)
 def continue_drum_sample(existing_audio_path):
@@ -265,7 +265,7 @@ with gr.Blocks() as iface:
             continue_output_audio = gr.Audio(label="Continued Music Output", type="filepath")
 
     # Connecting the components
-    generate_button.click(generate_drum_sample, outputs=[drum_audio])
+    # generate_button.click(generate_drum_sample, outputs=[drum_audio])
     continue_drum_sample_button.click(continue_drum_sample, inputs=[drum_audio], outputs=[drum_audio])
     generate_music_button.click(generate_music, inputs=[drum_audio, prompt_duration, musicgen_model, output_duration], outputs=[output_audio])
     continue_button.click(continue_music, inputs=[output_audio, prompt_duration, musicgen_model, output_duration], outputs=continue_output_audio)