Update app.py

app.py

@@ -28,8 +28,6 @@ st.subheader('Feeling like you might need a better coping strategy? Take the qui
 ###       data transformation      ###
 ### ------------------------------ ###
 
-global x_train, x_test, y_train, y_test
-
 # load dataset
 uncleaned_data = pd.read_csv('data.csv')
   
@@ -98,6 +96,8 @@ def train_model():
   # save the model to file
   with open('model.pkl', 'wb') as f:
     pkl.dump(model, f)
+    
+  return metrics.accuracy_score(y_test, y_pred)
 
 ### -------------------------------- ###
 ###            rerun logic           ###
@@ -107,7 +107,7 @@ try:
   with open('model.pkl', 'rb') as f:
     model = pkl.load(f)
 except FileNotFoundError as e:
-  train_model()
+  acc = train_model()
 
 
 
@@ -165,7 +165,7 @@ if form.form_submit_button("Submit to get your recommendation!"):
   
 col1, col2 = st.columns(2)
 col1.metric("Number of Options", len(cat_value_dicts[final_colname]))
-col2.metric("Model Accuracy", str(round(metrics.accuracy_score(y_test, y_pred) * 100, 1)) + '%')
+col2.metric("Model Accuracy", str(round(acc * 100, 1)) + '%')
 st.metric("Most Important Question", get_feat())