Update app.py

app.py

@@ -28,7 +28,7 @@ st.subheader('Feeling like you might need a better coping strategy? Take the qui
 ###       data transformation      ###
 ### ------------------------------ ###
 
-def load_dataset():
+def train_model():
   # load dataset
   uncleaned_data = pd.read_csv('data.csv')
   
@@ -73,13 +73,13 @@ def load_dataset():
   
     cat_value_dicts[colname] = new_dict
     data[colname] = transformed_col_vals
-
-
-### -------------------------------- ###
-###           model training         ###
-### -------------------------------- ###
-
-def train_model():
+    
+    
+  ### -------------------------------- ###
+  ###           model training         ###
+  ### -------------------------------- ###
+  
+  
   # select features and predicton; automatically selects last column as prediction
   cols = len(data.columns)
   num_features = cols - 1
@@ -105,7 +105,6 @@ try:
   with open('model.pkl', 'r') as f:
     model = pkl.load(f)
 except FileNotFoundError as e:
-  load_dataset()
   train_model()