Update app.py

app.py

@@ -1,24 +1,36 @@
 #!/usr/bin/env python3
-# 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
+# GhostAI Music Generator Hugging Face Spaces GPU-Compatible
+
+import spaces  # <--- Must be imported FIRST before torch and CUDA
+
+import os
+import sys
+import gc
+import warnings
+import tempfile
+import torch
+import torchaudio
+import numpy as np
+import 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")
 
+# Hugging Face token auth
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
     sys.exit("ERROR: HF_TOKEN not set.")
 login(HF_TOKEN)
 
+# Device setup
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Running on {device.upper()}")
 
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 
+# Load MusicGen model explicitly on GPU
 musicgen = MusicGen.get_pretrained("medium")
 musicgen.lm.to(device)
 musicgen.set_generation_params(duration=10)
@@ -28,15 +40,15 @@ def clean_resources():
         torch.cuda.empty_cache()
     gc.collect()
 
-@spaces.GPU  # <--- Required Hugging Face GPU decorator
+@spaces.GPU  # <-- Correct GPU decorator for HF
 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."
 
     sample_rate = musicgen.sample_rate
     segments = []
-
     chunks = max(1, total_len // chunk_len)
+
     for _ in range(chunks):
         with torch.no_grad():
             audio = musicgen.generate(