Update app.py

app.py

@@ -28,9 +28,10 @@ forecast_ngb_features = forecast_ngb_artifact["feature_names"]
 kmeans_artifact = joblib.load("kmeans_model.joblib")
 kmeans_model = kmeans_artifact["model"]
 cluster_scaler = kmeans_artifact["scaler"]
-cluster_features_names = kmeans_artifact["feature_names"]
-if "cluster" in cluster_features_names:
-    raise ValueError("cluster_features_names must NOT include 'cluster'")
+cluster_features_names = [
+    f for f in kmeans_artifact["feature_names"]
+    if f != "cluster"
+]