Update app.py

app.py

@@ -16,6 +16,8 @@ from os.path import exists
 ### ----------------------------- ###
 ###        interface setup        ###
 ### ----------------------------- ###
+global acc
+
 
 with open('styles.css') as f:
   st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
@@ -96,8 +98,10 @@ 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)
+  
+  acc = metrics.accuracy_score(y_test, y_pred)
+  
+  return model
 
 ### -------------------------------- ###
 ###            rerun logic           ###
@@ -107,7 +111,7 @@ try:
   with open('model.pkl', 'rb') as f:
     model = pkl.load(f)
 except FileNotFoundError as e:
-  acc = train_model()
+  model = train_model()