update

app.py

@@ -3,10 +3,12 @@ import torch
 from api import FlexSED
 import tempfile
 import os
+import spaces
 
 # Load model once on startup
 flexsed = FlexSED(device="cuda" if torch.cuda.is_available() else "cpu")
 
+@spaces.GPU  # 🚀 ensures this function runs on the GPU in Hugging Face Spaces
 def run_flexsed(audio_file, event_list):
     """
     Run inference using FlexSED and return prediction plot.

requirements.txt

@@ -1,6 +1,9 @@
 gradio
 torch
+torchaudio
 soundfile
 matplotlib
 numpy
-librosa
+librosa
+transformers
+einops