Update app.py

app.py

@@ -4,12 +4,12 @@ from transformers import pipeline
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 import scipy
 
-def generate_music():
+def generate_music(text):
     processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
     model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
     
     inputs = processor(
-        text=["80s pop track with bassy drums and synth", "90s rock song with loud guitars and heavy drums"],
+        text=text,
         padding=True,
         return_tensors="pt",
     )
@@ -21,7 +21,7 @@ def generate_music():
 demo = gr.Interface(
   fn=generate_music,
   inputs='text',
-  outputs='text',
+  outputs='file',
 )
 
 demo.launch()