Update app.py

app.py

@@ -19,7 +19,11 @@ def get_pipeline():
     if pipe is None:
         from yue2 import YuE2Pipeline
         print("Loading YuE2-3B pipeline on GPU...")
-        pipe = YuE2Pipeline.from_pretrained(REPO, device="cuda" if torch.cuda.is_available() else "cpu")
+        pipe = YuE2Pipeline.from_pretrained(
+            REPO, 
+            device="cuda" if torch.cuda.is_available() else "cpu",
+            trust_remote_code=True
+        )
         print("Pipeline loaded.")
     return pipe
 
@@ -30,7 +34,6 @@ def flac_to_mp3(flac_path, mp3_path, bitrate="320k"):
     )
     return mp3_path
 
-# ה-Decorator הקריטי שמפעיל את ה-ZeroGPU בזמן יצירת השיר
 @spaces.GPU(duration=120)
 def generate(style, lyrics, cot, seed, cfg_scale):
     pipeline = get_pipeline()
@@ -51,25 +54,85 @@ def generate(style, lyrics, cot, seed, cfg_scale):
     song.save(flac_path)
     flac_to_mp3(flac_path, mp3_path)
     
-    return flac_path, mp3_path, f"seed={seed} | {os.path.getsize(mp3_path) // 1024}KB MP3 320kbps"
+    return flac_path, mp3_path, f"Seed: {seed} | Size: {os.path.getsize(mp3_path) // 1024} KB (MP3 320kbps)"
 
-with gr.Blocks(title="YuE2-3B Music") as demo:
-    gr.Markdown("# YuE2-3B Music Generator\nFull song (vocals + accompaniment) from a style prompt and lyrics.")
+# CSS לעיצוב יוקרתי בסגנון Dark Theme
+custom_css = """
+body {
+    background-color: #0b0f19 !important;
+    color: #e2e8f0 !important;
+}
+.gradio-container {
+    max-width: 1100px !important;
+    margin: 0 auto !important;
+}
+.main-title {
+    text-align: center;
+    margin-bottom: 20px;
+}
+.main-title h1 {
+    font-size: 2.5rem;
+    font-weight: 800;
+    background: linear-gradient(90deg, #ff4b4b, #6b21a8);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+}
+.generate-btn {
+    background: linear-gradient(135deg, #ff4b4b 0%, #a855f7 100%) !important;
+    border: none !important;
+    color: white !important;
+    font-size: 1.1rem !important;
+    font-weight: bold !important;
+    border-radius: 12px !important;
+    padding: 12px !important;
+    box-shadow: 0 4px 15px rgba(255, 75, 75, 0.3) !important;
+    transition: all 0.3s ease !important;
+}
+.generate-btn:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 6px 20px rgba(255, 75, 75, 0.5) !important;
+}
+"""
+
+with gr.Blocks(css=custom_css, title="YuE2-3B Music Generator") as demo:
+    gr.HTML(
+        """
+        <div class="main-title">
+            <h1>🎵 YuE2-3B Music Studio</h1>
+            <p style="color: #94a3b8; font-size: 1.05rem;">Generate full songs with vocals and accompaniment directly on ZeroGPU</p>
+        </div>
+        """
+    )
+    
     with gr.Row():
-        with gr.Column():
-            style = gr.Textbox(label="Style prompt", value="Cyber metal, English, aggressive male vocals, distorted guitars, double-kick drums", lines=3)
-            lyrics = gr.Textbox(label="Lyrics ([verse]/[chorus] etc.)", value="[verse]\nnothing here but static skies\nwe ride the current till it dies", lines=8)
+        with gr.Column(scale=5):
+            style = gr.Textbox(
+                label="🎨 Style Prompt", 
+                value="Cyber metal, English, aggressive male vocals, distorted guitars, double-kick drums", 
+                lines=3
+            )
+            lyrics = gr.Textbox(
+                label="📜 Lyrics ([verse] / [chorus] structure)", 
+                value="[verse]\nnothing here but static skies\nwe ride the current till it dies", 
+                lines=8
+            )
             with gr.Row():
-                cot = gr.Radio(["full", "melody", "off"], value="full", label="CoT mode")
-                seed = gr.Number(label="Seed", value=0, precision=0)
-                cfg = gr.Number(label="CFG scale (1.0 = default)", value=1.0, precision=2)
-            btn = gr.Button("Generate song", variant="primary")
-        with gr.Column():
-            audio = gr.Audio(label="Play song", type="filepath")
-            mp3_file = gr.File(label="Download MP3 320kbps")
-            info = gr.Textbox(label="Info")
+                cot = gr.Radio(["full", "melody", "off"], value="full", label="CoT Mode")
+                seed = gr.Number(label="Seed (0 for Random)", value=0, precision=0)
+                cfg = gr.Number(label="CFG Scale", value=1.0, precision=2)
+                
+            btn = gr.Button("🚀 Generate Song", elem_classes=["generate-btn"])
             
-    btn.click(fn=generate, inputs=[style, lyrics, cot, seed, cfg], outputs=[audio, mp3_file, info], queue=True)
+        with gr.Column(scale=5):
+            audio = gr.Audio(label="🎧 Generated Song Player", type="filepath")
+            mp3_file = gr.File(label="📥 Download MP3 (320kbps)")
+            info = gr.Textbox(label="ℹ️ Generation Info", interactive=False)
+            
+    btn.click(
+        fn=generate, 
+        inputs=[style, lyrics, cot, seed, cfg], 
+        outputs=[audio, mp3_file, info], 
+        queue=True
+    )
 
-# הרצה תואמת Hugging Face Spaces
 demo.queue().launch()