Make the hardware a settings change, not a code change

app.py

@@ -17,6 +17,7 @@ from pyharp.tags import Subcategory
 
 from models.resunet import ResUNet30
 from models.clap_encoder import CLAP_Encoder
+import spaces
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -85,6 +86,7 @@ model_card = ModelCard(
 
 
 @torch.inference_mode()
+@spaces.GPU(duration=120)
 def process_fn(audio_path: str, text_query: str):
     if model_loading:
         raise gr.Error("Model is still loading, please wait a moment and try again.")