Update app.py

app.py

@@ -22,10 +22,22 @@ eccentricity = st.slider('Eccentricity', min_value=0.75693, max_value=0.81012, s
 extent = st.slider('Extent', min_value=0.656535, max_value=0.725739, step=0.01)
 convex_area = st.number_input('Convex hull (convex area)', min_value=18068.0, max_value=36683.0, step=0.01)
 
+# # Tombol untuk memprediksi
+# if st.button('Predict'):
+#     input_features = [[length_major_axis, width_minor_axis, thickness_depth, area,
+#                        perimeter, roundness, solidity, compactness, aspect_ratio,
+#                        eccentricity, extent, convex_area]]
+#     prediction = model.predict(input_features)
+#     st.write(f'The predicted class is: {prediction[0]}')
+
 # Tombol untuk memprediksi
 if st.button('Predict'):
     input_features = [[length_major_axis, width_minor_axis, thickness_depth, area,
                        perimeter, roundness, solidity, compactness, aspect_ratio,
                        eccentricity, extent, convex_area]]
     prediction = model.predict(input_features)
+    prediction_proba = model.predict_proba(input_features)
+
     st.write(f'The predicted class is: {prediction[0]}')
+    st.write(f'Prediction probabilities: {prediction_proba}')
+