Update app.py

app.py

@@ -4,8 +4,6 @@ import os
 import shutil
 import tempfile
 
-# is_shared_ui = True if "fffiloni/YuE" in os.environ['SPACE_ID'] else False
-
 # Install required package
 def install_flash_attn():
     try:
@@ -172,7 +170,6 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
         print("Temporary files deleted.")
 
 # Gradio 
-
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# YuE: Open Music Foundation Models for Full-Song Generation")
@@ -195,12 +192,8 @@ with gr.Blocks() as demo:
                 lyrics_txt = gr.Textbox(label="Lyrics")
                 
             with gr.Column():
-                if is_shared_ui:
-                    num_segments = gr.Number(label="Number of Segments", value=2, interactive=True)
-                    max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum="3000", step=500, value=1500, interactive=True)
-                else:
-                    num_segments = gr.Number(label="Number of Song Segments", value=2, interactive=True)
-                    max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum="24000", step=500, value=3000, interactive=True)
+                num_segments = gr.Number(label="Number of Song Segments", value=2, interactive=True)
+                max_new_tokens = gr.Slider(label="Max New Tokens", minimum=500, maximum=24000, step=500, value=3000, interactive=True)
                 submit_btn = gr.Button("Submit")
                 music_out = gr.Audio(label="Audio Result")