Update app.py

app.py

@@ -1,13 +1,6 @@
 import gradio as gr
-from diffusers import DiffusionPipeline
 
-model_name = "facebook/musicgen-small"  
-pipe = DiffusionPipeline.from_pretrained(model_name)
-
-def generate(prompt):
-    audio = pipe(prompt)
-    return audio
-
-demo = gr.Interface(fn = generate, inputs="text", outputs=audio.save("output_music.wav"))
-
-demo.launch()
+gr.load(
+   "models/facebook/musicgen-small",
+   provider="hf-inference",
+).launch()