Adapted this code from fffiloni/YuE

app.py

@@ -3,8 +3,9 @@ import subprocess
 import os 
 import shutil
 import tempfile
+import spaces
 
-is_shared_ui = True if "/" in os.environ['SPACE_ID'] else False
+is_shared_ui = True if "innova-ai/YuE-music-generator-demo" in os.environ['SPACE_ID'] else False
 
 # Install required package
 def install_flash_attn():
@@ -101,6 +102,7 @@ def get_last_mp3_file(output_dir):
     # Return the most recent .mp3 file
     return mp3_files_with_path[0]
 
+@spaces.GPU
 def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
     # Create temporary files
     genre_txt_path = create_temp_file(genre_txt_content, prefix="genre_")
@@ -184,7 +186,7 @@ with gr.Blocks() as demo:
             <a href="https://map-yue.github.io">
                 <img src='https://img.shields.io/badge/Project-Page-green'>
             </a>
-            <a href="https://huggingface.co/spaces/fffiloni/YuE?duplicate=true">
+            <a href="https://huggingface.co/spaces/innova-ai/YuE-music-generator-demo?duplicate=true">
                 <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
             </a>
         </div>