Update app.py

app.py

@@ -1,10 +1,11 @@
 #!/usr/bin/env python3
-# GhostAI Music Generator Hugging Face Spaces Version
+# GhostAI Music Generator Hugging Face Spaces GPU Compatible Version
 import os, sys, gc, time, warnings, tempfile
 import torch, torchaudio, numpy as np, gradio as gr
 from pydub import AudioSegment
 from audiocraft.models import MusicGen
 from huggingface_hub import login
+import spaces  # Hugging Face spaces decorator
 
 warnings.filterwarnings("ignore")
 
@@ -16,7 +17,6 @@ login(HF_TOKEN)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Running on {device.upper()}")
 
-# Fix transformers compatibility manually
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 
 musicgen = MusicGen.get_pretrained("medium")
@@ -28,6 +28,7 @@ def clean_resources():
         torch.cuda.empty_cache()
     gc.collect()
 
+@spaces.GPU  # <--- Required Hugging Face GPU decorator
 def generate_music(prompt, cfg, top_k, top_p, temp, total_len, chunk_len, crossfade):
     if not prompt.strip():
         return None, "⚠️ Enter a valid prompt."