remove spaces.GPU decorator from load_model function and update requirements to include cuda-python

app.py

@@ -1,7 +1,6 @@
 import os
 import gradio as gr
 import torch
-import spaces
 import nemo.collections.asr as nemo_asr
 from omegaconf import OmegaConf
 import time
@@ -11,8 +10,7 @@ print(f"CUDA available: {torch.cuda.is_available()}")
 if torch.cuda.is_available():
     print(f"CUDA device: {torch.cuda.get_device_name(0)}")
 
-# Initialize the ASR model
-@spaces.GPU
+# Initialize the ASR model - removed spaces.GPU decorator due to pickling issues
 def load_model():
     print("Loading ASR model...")
     # Load the NVIDIA NeMo ASR model

requirements.txt

@@ -2,5 +2,5 @@ torch>=1.13.0
 gradio>=3.32.0
 nemo_toolkit[asr]>=1.18.0
 omegaconf>=2.2.0
-spaces>=0.15.0
 numpy>=1.22.0
+cuda-python>=12.3