add

app.py

@@ -499,5 +499,11 @@ with gr.Blocks(title="MusicGen Large - Music Generation") as demo:
     """)
 
 if __name__ == "__main__":
+    try:
+        subprocess.run(['bash', 'setup.sh.sh'], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error: {e}")
+    except FileNotFoundError:
+        print("Shell script not found")
     cleanup_temp_files()
     demo.launch()

setup.sh

@@ -6,3 +6,5 @@ pip install flash-attn==2.7.3 --no-build-isolation
 
 # Uninstall apex if it exists (as in your original script)
 pip uninstall apex -y || true
+
+thestage config set --api-token eyJhbGciOiJIUzI1NiJ9.eyJqdGkiOiI1ODFmMDdlZi1kN2U0LTQ2ZTUtODIxZS0xNDBhNmQ4YjJkZWIiLCJzdWIiOiJhcGlVc2VyIiwidXNlcklkIjoyMiwiY2xpZW50SWQiOjMzLCJ0b2tlblR5cGUiOiJhY2Nlc3MiLCJleHAiOjE3NzkxOTc2NDh9.lwb44HoHP7n9WZ7ZUux7Bw9RqsQCwKqgvWvdizMnfRE