Update app.py

app.py

@@ -984,17 +984,19 @@ 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
 elif app_mode == "Visualization Lab":