Only the call to GPU

app.py

@@ -92,7 +92,7 @@ def load_model_if_needed():
     
     return MODEL
 
-
+@spaces.GPU
 def prompt_lm(audios: list[str], messages: list[dict[str, str]]):
     # Always try to load the model if needed
     model = load_model_if_needed()