Update app.py

app.py

@@ -101,11 +101,13 @@ def train_model():
 ###            rerun logic           ###
 ### -------------------------------- ###
 
-if not open('model.pkl'):
+try:
+  model = pkl.load('model.pkl')
+except FileNotFoundError as e:
   load_dataset()
   train_model()
 
-model = pkl.load('model.pkl')
+
 
 
 ### ------------------------------- ###