Update pages/5_RealDataSetRegression.py

pages/5_RealDataSetRegression.py

@@ -73,6 +73,8 @@ batch_size = st.sidebar.slider('Batch Size', min_value=1, max_value=len(training
 feature = st.sidebar.selectbox('Select Feature', training_df.columns)
 label = 'median_house_value'
 
+my_model = None  # Initialize the model variable
+
 if st.sidebar.button('Run'):
     my_model = build_model(learning_rate)
     weight, bias, epochs, rmse = train_model(my_model, training_df, feature, label, epochs, batch_size)
@@ -98,6 +100,6 @@ def predict_house_values(n, feature, label):
                                         predicted_values[i][0] ))
 
 n_predictions = st.sidebar.slider('Number of Predictions', min_value=1, max_value=100, value=10)
-if st.sidebar.button('Predict'):
+if my_model is not None and st.sidebar.button('Predict'):
     st.subheader('Predictions')
     predict_house_values(n_predictions, feature, label)