adapted script for zero gpu

app.py

@@ -1,4 +1,5 @@
 import argparse
+import spaces
 import sys
 from pathlib import Path
 
@@ -14,7 +15,7 @@ from sfi_utmos.model.ssl_mos import SSLMOSLightningModule
 
 # Global variable for the model
 model: SSLMOSLightningModule | None = None
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda"
 
 
 def load_model(checkpoint_path: str):
@@ -28,7 +29,7 @@ def load_model(checkpoint_path: str):
     model.eval()
     print(f"Model loaded from {checkpoint_path}")
 
-
+@spaces.GPU
 def predict_mos(audio_path: str):
     """Predicts the MOS score for the given audio file."""
     if model is None: