no more duration params

app.py

@@ -18,8 +18,8 @@ def preprocess_audio(waveform):
     waveform_np = waveform.cpu().squeeze().numpy()
     return torch.from_numpy(waveform_np).unsqueeze(0).to(device)
 
-@spaces.GPU  # Remove duration parameter
-def generate_drum_sample():
+@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)
@@ -31,7 +31,7 @@ def generate_drum_sample():
 
     return filename_with_extension
 
-@spaces.GPU  # Remove duration parameter
+@spaces.GPU(duration=10)
 def continue_drum_sample(existing_audio_path):
     if existing_audio_path is None:
         return None
@@ -69,7 +69,7 @@ def continue_drum_sample(existing_audio_path):
 
     return combined_file_path
 
-@spaces.GPU  # Remove duration parameter
+@spaces.GPU(duration=120)
 def generate_music(wav_filename, prompt_duration, musicgen_model, output_duration):
     if wav_filename is None:
         return None
@@ -104,7 +104,7 @@ def generate_music(wav_filename, prompt_duration, musicgen_model, output_duratio
 
     return filename_with_extension
 
-@spaces.GPU  # Remove duration parameter
+@spaces.GPU(duration=120)
 def continue_music(input_audio_path, prompt_duration, musicgen_model, output_duration):
     if input_audio_path is None:
         return None
@@ -166,7 +166,7 @@ def continue_music(input_audio_path, prompt_duration, musicgen_model, output_dur
 
     return combined_audio_filename
 
-# Define the expandable sections
+# Define the expandable sections (keeping your existing content)
 musicgen_micro_blurb = """
 ## musicgen_micro
 musicgen micro is an experimental series of models by aaron abebe. they are incredibly fast, and extra insane. this one does goated jungle drums. we're very excited about these.
@@ -209,7 +209,7 @@ thepatch/PhonkV2 was trained by MJ BERSABEph. there are multiple versions in the
 foureyednymph/musicgen-sza-sos-small was just trained by foureyednymph. We're all about to find out if it does continuations well.
 """
 
-# Create the Gradio interface
+# Create the Gradio interface with explicit types
 with gr.Blocks() as iface:
     gr.Markdown("# the-micro-slot-machine")
     gr.Markdown("two ai's jamming. warning: outputs will be very strange, likely stupid, and possibly rad.")