Update app.py

app.py

@@ -7,6 +7,7 @@ import json
 import gradio as gr
 from omegaconf import OmegaConf
 from huggingface_hub import hf_hub_download
+import spaces
 
 # Add the SongBloom module to the path
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
@@ -76,6 +77,7 @@ class SongBloomApp:
             self.is_loading = False
             return f"Error loading model: {str(e)}"
 
+    @spaces.GPU
     def generate_song(self, lyrics, prompt_audio, n_samples=1, dtype="float32", progress=gr.Progress()):
         """Generate song from lyrics and audio prompt"""
         if self.model is None: