ok herewego

app.py

@@ -20,24 +20,22 @@ def preprocess_audio(waveform):
 
 # Test with a wrapper function
 @spaces.GPU
-def generate_drum_sample(progress=gr.Progress(track_tqdm=True)):
-    progress(0.1, desc="Loading model...")
+def _generate_drum_sample_internal():
     model = MusicGen.get_pretrained('pharoAIsanders420/micro-musicgen-jungle')
-    
-    progress(0.3, desc="Setting parameters...")
     model.set_generation_params(duration=10)
-    
-    progress(0.5, desc="Generating audio...")
     wav = model.generate_unconditional(1).squeeze(0)
     
-    progress(0.8, desc="Saving file...")
     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)
-    
-    progress(1.0, desc="Complete!")
+
     return filename_with_extension
 
+# Regular function wrapper (no @spaces.GPU on this one)
+def generate_drum_sample():
+    return _generate_drum_sample_internal()
+
 
 
 # @spaces.GPU
@@ -273,12 +271,14 @@ with gr.Blocks() as iface:
             continue_button = gr.Button("Continue Generating Music")
             continue_output_audio = gr.Audio(label="Continued Music Output", type="filepath")
 
-    # Hidden component to provide dummy input
-    hidden_trigger = gr.Textbox(value="generate", visible=False)
-
     # Fixed click handlers - use hidden input for generate_drum_sample
     # Normal click connection
-    generate_button.click(generate_drum_sample, outputs=[drum_audio])
+    generate_button.click(
+        fn=generate_drum_sample,
+        inputs=None,
+        outputs=drum_audio,
+        api_name="generate"  # Add explicit API name
+    )
     # 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)