Add spaces GPU decorator to use zeroGPU resources

app.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass, field
 import logging
+import spaces
 
 import sys
 sys.path.append("/home/user/app/src/sonicverse")
@@ -50,6 +51,7 @@ model, tokenizer = load_trained_lora_model(
     tasks_config=serve_args.tasks_config
 )
 
+@spaces.GPU(duration=300)
 def caption_audio(audio_file):
     chunk_audio_files = split_audio(audio_file, CHUNK_LENGTH)
     chunk_captions = []