Deploy Music Generation Studio - 2025-12-12 17:35

README.md

@@ -9,7 +9,6 @@ app_file: app.py
 pinned: false
 license: mit
 python_version: 3.11
-suggested_hardware: cpu-basic
 ---
 
 # 🎵 Music Generation Studio

app.py

@@ -10,6 +10,18 @@ from pathlib import Path
 import shutil
 import subprocess
 
+# Import spaces for ZeroGPU support
+try:
+    import spaces
+    HAS_SPACES = True
+except ImportError:
+    HAS_SPACES = False
+    # Create a dummy decorator for local development
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            return func
+
 # Run DiffRhythm2 source setup if needed
 setup_script = Path(__file__).parent / "setup_diffrhythm2_src.sh"
 if setup_script.exists():
@@ -80,6 +92,7 @@ def get_lyricmind_service():
         logger.info("LyricMind model loaded")
     return lyricmind_service
 
+@spaces.GPU
 def generate_lyrics(prompt: str, duration: int, progress=gr.Progress()):
     """Generate lyrics from prompt using analysis"""
     try:
@@ -112,6 +125,7 @@ def generate_lyrics(prompt: str, duration: int, progress=gr.Progress()):
         logger.error(f"Error generating lyrics: {e}", exc_info=True)
         return f"❌ Error: {str(e)}"
 
+@spaces.GPU
 def generate_music(prompt: str, lyrics: str, lyrics_mode: str, duration: int, position: str, progress=gr.Progress()):
     """Generate music clip and add to timeline"""
     try:

requirements.txt

@@ -1,5 +1,6 @@
 # Core dependencies for HuggingFace Spaces deployment
 gradio==4.44.0
+spaces>=0.28.3
 numpy>=1.24.0,<2.0.0
 scipy>=1.10.0
 librosa>=0.10.0
@@ -7,7 +8,7 @@ soundfile>=0.12.0
 pydantic>=2.0.0
 pyyaml>=6.0
 
-# PyTorch - CPU mode for HuggingFace Spaces
+# PyTorch - GPU mode for ZeroGPU
 torch>=2.4.0,<2.5.0
 torchaudio>=2.4.0,<2.5.0