Update app.py

app.py

@@ -1,9 +1,14 @@
 from transformers import pipeline
 
-pipe = pipeline("text-to-music", model="openai/jukebox")
+from transformers import AutoProcessor, MusicgenForConditionalGeneration
+
+processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
+model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
+
+
 
 def generate(description):
-    audio = pipe(description)
+    audio = processor(description, padding=True, return_tensors="pt")
     return audio['audio']
 
 demo = gr.Interface(