Update app.py

app.py

@@ -11,9 +11,14 @@ if HUGGINGFACE_TOKEN is None:
     raise ValueError("Missing Hugging Face token. Please set it in Spaces Secrets.")
 login(HUGGINGFACE_TOKEN)
 
-# Set device for PyTorch
-device = "cuda" if torch.cuda.is_available() else "cpu"
-torch_dtype = torch.float16 if device == "cuda" else torch.float32
+# Set device for PyTorch (only CPU, if no GPU is available)
+device = "cpu"
+torch_dtype = torch.float32  # Use float32 for CPU by default
+
+# Check for GPU availability
+if torch.cuda.is_available():
+    device = "cuda"
+    torch_dtype = torch.float16  # Use float16 for GPU to optimize memory usage
 
 # Load the pipeline
 pipe = StableAudioPipeline.from_pretrained(