Update app.py

app.py

@@ -106,6 +106,7 @@ class KerasClassifierWrapper(BaseEstimator, ClassifierMixin):
     def predict(self, X):
         return (self.model.predict(X) > 0.5).astype(int).flatten()
 
+wrapper_model = None
 if st.session_state.running:
     model = build_ann()
     history = model.fit(X_train, y_train, epochs=num_epochs, batch_size=batch_size, validation_data=(X_test, y_test), verbose=0)