Update app.py

app.py

@@ -67,14 +67,18 @@ def visualize_knn(n_neighbors):
     return temp_file.name, f"Accuracy: {acc:.4f}"
 
 
+# -------- CSS FIX (Gradio 6 requires CSS in launch()) ------
 custom_css = """
-.gr-button {
+button, .gr-button, .gr-button-primary, .gr-btn {
     background-color: #007bff !important;
     color: white !important;
     border-radius: 8px !important;
     padding: 12px 20px !important;
     font-weight: bold !important;
 }
+button:hover {
+    background-color: #0056cc !important;
+}
 .gr-slider input {
     accent-color: #007bff !important;
 }
@@ -87,7 +91,9 @@ body, .gradio-container {
 }
 """
 
-with gr.Blocks(css=custom_css) as demo:
+
+# -------------------- GRADIO UI --------------------
+with gr.Blocks() as demo:
 
     gr.Markdown("## 🧠 KNN Decision Boundary + Dynamic Train/Test Split Visualizer")
 
@@ -108,4 +114,4 @@ with gr.Blocks(css=custom_css) as demo:
     visualize_btn.click(visualize_knn, inputs=[k_slider], outputs=[output_img, accuracy_text])
 
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name="0.0.0.0", server_port=7860, css=custom_css)