Spaces:

MaxGab
/

Text-to-Audio

Running

MaxGab commited on Mar 20

Commit

d5d113f

verified ·

1 Parent(s): 02c23b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,8 @@ model_name = "facebook/musicgen-small"
 pipe = DiffusionPipeline.from_pretrained(model_name)
 def generate(prompt):
-    return audio = pipe(prompt)
 demo = gr.Interface(fn = generate, inputs="text", outputs=audio.save("output_music.wav"))

 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"))