Update app.py

app.py

@@ -19,8 +19,25 @@ from transformers import pipeline
 import click
 import soundfile as sf
 
-# Remove GPU decorator and Spaces import
-device = "cpu"
+try:
+    import spaces
+    USING_SPACES = True
+except ImportError:
+    USING_SPACES = False
+
+def gpu_decorator(func):
+    if USING_SPACES:
+        return spaces.GPU(func)
+    else:
+        return func
+
+device = (
+    "gpu"
+    if torch.cuda.is_available()
+    else "mps" if torch.backends.mps.is_available() else "cpu"
+)
+
+print(f"Using {device} device")
 print(f"Using {device} device")
 
 # Use smaller, CPU-friendly models
@@ -34,11 +51,11 @@ vocos = Vocos.from_pretrained("charactr/vocos-mel-24khz")
 
 # Reduce computational intensity
 target_sample_rate = 24000
-n_mel_channels = 100  # Reduced from 100
+n_mel_channels = 80  # Reduced from 100
 hop_length = 256  # Increased from 256
 target_rms = 0.1
 nfe_step = 12  # Reduced from 8
-cfg_strength = 2  # Reduced from 2.0
+cfg_strength = 1.5  # Reduced from 2.0
 ode_method = "euler"
 sway_sampling_coef = -1
 speed = 1