Update model_utils.py

model_utils.py

@@ -2,12 +2,14 @@ import torch
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 import whisper
 from config import WHISPER_MODEL_SIZE
+import spaces 
 
 # Global variables to store models
 whisper_processor = None
 whisper_model = None
 whisper_model_small = None
 
++@spaces.GPU
 def load_models():
     global whisper_processor, whisper_model, whisper_model_small
     if whisper_processor is None:
@@ -17,21 +19,25 @@ def load_models():
     if whisper_model_small is None:
         whisper_model_small = whisper.load_model(WHISPER_MODEL_SIZE)
 
++@spaces.GPU
 def get_device():
     return "cuda:0" if torch.cuda.is_available() else "cpu"
 
++@spaces.GPU
 def get_processor():
     global whisper_processor
     if whisper_processor is None:
         load_models()
     return whisper_processor
 
++@spaces.GPU
 def get_model():
     global whisper_model
     if whisper_model is None:
         load_models()
     return whisper_model
 
++@spaces.GPU
 def get_whisper_model_small():
     global whisper_model_small
     if whisper_model_small is None: