add space gpu

app.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 import gradio as gr
 
 from transformers import MusicgenForConditionalGeneration
@@ -11,10 +12,21 @@ model.to(device)
 from transformers import AutoProcessor
 processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
 
-
+@spaces.GPU
 def generate_music(desc):
     inputs = processor(text=[desc], padding=True, return_tensors="pt")
     audio_values = model.generate(**inputs.to(device), do_sample=True, guidance_scale=3, max_new_tokens=256)
     return sampling_rate, audio_values[0][0].cpu().numpy()
     
-gr.Interface(fn=generate_music, inputs="text", outputs="audio").launch()
+# gr.Interface(fn=generate_music, inputs="text", outputs="audio").launch()
+
+with gr.Blocks() as app:
+    with gr.Row():
+        music_desc = gr.TextArea(label="Music Description")
+        music_player = gr.Audio(label="Play My Tune")
+
+    gen_btn = gr.Button("Get Some Tune!!")
+    gen_btn.click(fn=generate_music, inputs=[music_desc], outputs=[music_player])
+    
+if __name__ == '__main__':
+    app.launch()

requirements.txt

@@ -2,3 +2,4 @@ transformers
 torch
 torchvision
 torchaudio
+spaces