Add support for GPU acceleration in sound generation and update dependencies

app.py

@@ -91,4 +91,4 @@ def home(request: Request):
 
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860, workers=2)

requirements.txt

@@ -8,3 +8,5 @@ pydantic==2.1.1
 fastapi==0.115.11
 uvicorn==0.33.0
 gradio==5.20.1
+spaces==0.32.0
+

sound_generator.py

@@ -1,5 +1,6 @@
 import time
 
+import spaces
 import torch
 from audiocraft.data.audio import audio_write
 from audiocraft.models import AudioGen, MusicGen
@@ -15,6 +16,7 @@ music_model = MusicGen.get_pretrained('facebook/musicgen-small')
 sound_model.set_generation_params(duration=5)
 music_model.set_generation_params(duration=5)
 
+@spaces.GPU
 def generate_sound(prompt: str):
     """
     Generate sound using Audiocraft based on the given prompt.
@@ -34,6 +36,7 @@ def generate_sound(prompt: str):
     
     return f"{output_path}.wav"
 
+@spaces.GPU
 def generate_music(prompt: str):
     """
     Generate music using Audiocraft based on the given prompt.