add spaces import and reintroduce GPU decorator for model loading

app.py

@@ -4,6 +4,7 @@ import torch
 import nemo.collections.asr as nemo_asr
 from omegaconf import OmegaConf
 import time
+import spaces
 
 # Check if CUDA is available
 print(f"CUDA available: {torch.cuda.is_available()}")
@@ -11,12 +12,14 @@ if torch.cuda.is_available():
     print(f"CUDA device: {torch.cuda.get_device_name(0)}")
 
 # Initialize the ASR model - removed spaces.GPU decorator due to pickling issues
+@spaces.GPU
 def load_model():
     print("Loading ASR model...")
     # Load the NVIDIA NeMo ASR model
     model = nemo_asr.models.EncDecRNNTBPEModel.from_pretrained("nvidia/parakeet-tdt-0.6b-v2")
     # Move model to GPU if available
     if torch.cuda.is_available():
+        print(f"CUDA device: {torch.cuda.get_device_name(0)}")
         model = model.cuda()
     print(f"Model loaded on device: {model.device}")
     return model