update

app.py

@@ -222,23 +222,23 @@ X_test = pd.DataFrame(X_test, columns=feature_names)
 if 'models_trained' not in st.session_state:
     st.session_state['models_trained'] = False
 
-# ML Models
-st.subheader("Machine Learning Models")
-
-# Initialize session state for models
-if 'models' not in st.session_state:
-    st.session_state['models'] = {
-        "Logistic Regression": LogisticRegression(),
-        "Naive Bayes": GaussianNB(),
-        "SVM": SVC(),
-        "Decision Tree": DecisionTreeClassifier(),
-        "Random Forest": RandomForestClassifier(),
-        "Gradient Boosting": GradientBoostingClassifier(),
-        "MLP Neural Network": MLPClassifier()
-    }
-
-# Create tabs for different models
-model_tabs = st.tabs(st.session_state['models'].keys())
+    # ML Models
+    st.subheader("Machine Learning Models")
+
+    # Initialize session state for models
+    if 'models' not in st.session_state:
+        st.session_state['models'] = {
+            "Logistic Regression": LogisticRegression(),
+            "Naive Bayes": GaussianNB(),
+            "SVM": SVC(),
+            "Decision Tree": DecisionTreeClassifier(),
+            "Random Forest": RandomForestClassifier(),
+            "Gradient Boosting": GradientBoostingClassifier(),
+            "MLP Neural Network": MLPClassifier()
+        }
+
+    # Create tabs for different models
+    model_tabs = st.tabs(st.session_state['models'].keys())
 
 # Train the models and store them in session state
 if not st.session_state['models_trained']: