Update app.py

app.py

@@ -47,17 +47,7 @@ if selected_features:
 
     # Prediction
     y_pred = model.predict(X_test)
-
-    # Display results
-    st.write("Selected Features", selected_features)
-    st.write("Model Coefficients", model.coef_)
-    st.write("Predictions", y_pred)
-    st.write("Actual Values", y_test.values)
-
-    # Model performance
-    st.write("Mean Squared Error", mean_squared_error(y_test, y_pred))
-    st.write("R-squared Score", r2_score(y_test, y_pred))
-
+    
     # Plot
     fig, ax = plt.subplots()
     ax.scatter(y_test, y_pred, edgecolors=(0, 0, 0))
@@ -66,3 +56,13 @@ if selected_features:
     ax.set_ylabel('Predicted')
     ax.set_title('Actual vs Predicted')
     st.pyplot(fig)
+
+    # Model performance
+    st.write("Mean Squared Error", mean_squared_error(y_test, y_pred))
+    st.write("R-squared Score", r2_score(y_test, y_pred))
+
+    # Display results
+    st.write("Selected Features", selected_features)
+    st.write("Model Coefficients", model.coef_)
+    st.write("Predictions", y_pred)
+    st.write("Actual Values", y_test.values)