add @spaces.GPU

app.py

@@ -5,6 +5,7 @@ from pathlib import Path
 from typing import Literal
 
 import gradio as gr
+import spaces
 import torch
 
 from NatureLM.models.NatureLM import NatureLM
@@ -27,6 +28,7 @@ class DummyConfig:
         }
 
 
+@spaces.GPU
 def prompt_lm(audios: list[str], messages: list[dict[str, str]]):
     cuda_enabled = torch.cuda.is_available()
     samples = prepare_sample_waveforms(audios, cuda_enabled)
@@ -213,6 +215,7 @@ def to_raven_format(outputs: dict[int, str], chunk_len: int = 10) -> str:
     return "\n".join(raven_output)
 
 
+@spaces.GPU
 def _run_long_recording_inference(file, task, chunk_len: int = 10, hop_len: int = 5, progress=gr.Progress()):
     cuda_enabled = torch.cuda.is_available()
     outputs = {}