Update app.py

app.py

@@ -28,6 +28,7 @@ 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')
@@ -87,7 +88,7 @@ def train_model():
   y = data.iloc[: , num_features:]
   
   # split data into training and testing sets
-  global x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.25)
+  x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.25)
   
   # instantiate the model (using default parameters)
   model = LogisticRegression()