Update app.py

app.py

@@ -11,7 +11,7 @@ from transformers.generation.streamers import BaseStreamer
 import gradio as gr
 import io, wave
 
-# import spaces
+import spaces
 
 
 from transformers import MusicgenMelodyForConditionalGeneration, MusicgenForConditionalGeneration, AutoProcessor, set_seed
@@ -607,7 +607,7 @@ def wave_header_chunk(frame_input=b"", channels=1, sample_width=2, sample_rate=2
     
     return wav_buf.read()
 
-# @spaces.GPU()
+@spaces.GPU()
 def generate_audio(text_prompt, audio, audio_length_in_s=10.0, play_steps_in_s=2.0, seed=0):
     max_new_tokens = int(frame_rate * audio_length_in_s)
     play_steps = int(frame_rate * play_steps_in_s)