ok herewego

app.py

@@ -20,22 +20,24 @@ def preprocess_audio(waveform):
 
 # Test with a wrapper function
 @spaces.GPU
-def _generate_drum_sample_internal():
+def generate_drum_sample(progress=gr.Progress(track_tqdm=True)):
+    progress(0.1, desc="Loading model...")
     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
@@ -276,10 +278,7 @@ with gr.Blocks() as iface:
 
     # Fixed click handlers - use hidden input for generate_drum_sample
     # Normal click connection
-    generate_button.click(
-        lambda: 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)