Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import numpy as np
 import soundfile as sf
 import torch
+import spaces
 
 from dia.model import Dia
 
@@ -21,21 +22,6 @@ parser.add_argument("--share", action="store_true", help="Enable Gradio sharing"
 
 args = parser.parse_args()
 
-
-# Determine device
-if args.device:
-    device = torch.device(args.device)
-elif torch.cuda.is_available():
-    device = torch.device("cuda")
-# Simplified MPS check for broader compatibility
-elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
-    # Basic check is usually sufficient, detailed check can be problematic
-    device = torch.device("mps")
-else:
-    device = torch.device("cpu")
-
-print(f"Using device: {device}")
-
 # Load Nari model and config
 print("Loading Nari model...")
 try:
@@ -46,6 +32,7 @@ except Exception as e:
     raise
 
 
+@spaces.GPU
 def run_inference(
     text_input: str,
     audio_prompt_input: Optional[Tuple[int, np.ndarray]],