Update app.py

app.py

@@ -540,8 +540,8 @@ with gr.Blocks(css="""
 
         # Explanation section with creative spacing and minimal design
         gr.Markdown("""
-            <div style="background-color: #f0f0f0; padding: 15px; border-radius: 10px; color: #333;">
-              <h3 style="color: #0056b3;">📡 <b>Beam Prediction Task</b></h3>
+            <div style="background-color: var(--primary-background); padding: 15px; border-radius: 10px; color: var(--text-primary);">
+              <h3 style="color: var(--text-primary);">📡 <b>Beam Prediction Task</b></h3>
               <ul style="padding-left: 20px;">
                 <li><b>🎯 Goal</b>: Predict the strongest <b>mmWave beam</b> from a predefined codebook using Sub-6 GHz channels.</li>
                 <li><b>⚙️ Adjust Settings</b>: Use the sliders to control the training data percentage and task complexity (beam count) to explore model performance.</li>