Update app.py

app.py

@@ -5,31 +5,31 @@ from audiocraft.data.audio import audio_write
 import spaces
 
 @spaces.GPU(duration=120)
+@spaces.GPU(duration=120) 
 def generate_music(description, melody_audio):
+
     model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
-    model.set_generation_params(duration=8)  # generate 8 seconds.
+    model.set_generation_params(duration=8)  
 
     if description:
-        descriptions = [description] * 3
+        description = [description]
         if melody_audio:
             melody, sr = torchaudio.load(melody_audio)
-            wav = model.generate_with_chroma(descriptions, melody[None].expand(3, -1, -1), sr)
+            wav = model.generate_with_chroma(description, melody[None], sr) 
         else:
-            wav = model.generate(descriptions)  # generates 3 samples.
+            wav = model.generate(description)  
     else:
-        wav = model.generate_unconditional(4)    # generates 4 unconditional audio samples
+        wav = model.generate_unconditional(1)  
 
-    output_audios = []
-    for idx, one_wav in enumerate(wav):
-        output_path = f'output_{idx}.wav'
-        audio_write(output_path, one_wav.cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
-        output_audios.append(output_path)
+    output_path = 'output.wav'
+    audio_write(output_path, wav[0].cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
+    
+    return output_path
 
-    return output_audios
 
 description = gr.Textbox(label="Description", placeholder="acoustic, guitar, melody, trap, d minor, 90 bpm")
 melody_audio = gr.Audio(label="Melody Audio (optional)", type="filepath")
-output_audio = gr.Audio(label="Generated Music", type="filepath")
+output_path = gr.Audio(label="Generated Music", type="filepath")
 
 gr.Interface(
     fn=generate_music,