Update app.py

app.py

@@ -22,6 +22,8 @@ import joblib
 import io, base64
 import json
 
+K_OPTIMAL = 5
+
 # Load saved artifacts
 kmeans_loaded   = joblib.load('kmeans_model.pkl')
 scaler_loaded   = joblib.load('scaler.pkl')
@@ -115,6 +117,8 @@ def predict_segment(age, annual_income, spending_score):
     user_scaled = scaler_loaded.transform(user_input)
     cluster_id = int(kmeans_loaded.predict(user_scaled)[0])
 
+    K_OPTIMAL = 5
+
     info = insights_loaded[cluster_id]
     color = SEGMENT_COLORS[cluster_id]
     emoji = SEGMENT_EMOJIS[cluster_id]