Update app.py

app.py

@@ -35,18 +35,14 @@ new_data = np.array([[browsing_frequency_value, shopping_satisfaction]])
 # Melakukan prediksi kluster dengan model yang dilatih
 predicted_cluster = model.predict(new_data)[0]
 
-# Pastikan predicted_cluster adalah integer
-# predicted_cluster = int(predicted_cluster)
-
 # Mapping kluster ke nama cluster
-# cluster_mapping = {
-#     0: 'cluster1',
-#     1: 'cluster2',
-#     2: 'cluster3',
-#     3: 'cluster4'
-# }
-# cluster_name = cluster_mapping[predicted_cluster]
+cluster_mapping = {
+    0: 'cluster1',
+    1: 'cluster2',
+    2: 'cluster3',
+    3: 'cluster4'
+}
+cluster_name = cluster_mapping[predicted_cluster]
 
 # Menampilkan hasil prediksi
-# st.write(f"Prediksi kluster untuk data yang diberikan adalah: {cluster_name}")
-st.write(f"Prediksi kluster untuk data yang diberikan adalah: {predicted_cluster}")
+st.write(f"Prediksi kluster untuk data yang diberikan adalah: {cluster_name}")