Update app.py

app.py

@@ -984,18 +984,17 @@ elif app_mode == "Model Training":
             except Exception as e:
                 st.error(f"An error occurred during training: {e}")
 
-             if st.session_state.model is not None:
-                st.subheader("💾 Save Model")
-                model_filename = st.text_input("Enter Model Filename (without extension)", "trained_model")
-    
-            if st.button("Save Model"):
-                try:
-                    joblib.dump(st.session_state.model, f"{model_filename}.joblib")
-                    st.success(f"Model saved as {model_filename}.joblib")
-                except Exception as e:
-                    st.error(f"Error saving model: {e}")
-        else:
-            st.warning("No trained model available. Train a model first to enable saving.")
+    if st.session_state.model is not None:
+        st.subheader("💾 Save Model")
+        model_filename = st.text_input("Enter Model Filename (without extension)", "trained_model")
+        if st.button("Save Model"):
+            try:
+                joblib.dump(st.session_state.model, f"{model_filename}.joblib")
+                st.success(f"Model saved as {model_filename}.joblib")
+            except Exception as e:
+                st.error(f"Error saving model: {e}")
+    else:
+        st.warning("No trained model available. Train a model first to enable saving.")
 
 
 # Visualization Lab Section