Isabel Gwara commited on
Commit
b004973
·
1 Parent(s): f100c9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -95,7 +95,8 @@ def train_model():
95
  y_pred = model.predict(x_test)
96
 
97
  # save the model to file
98
- pkl.dump(model, 'model.pkl')
 
99
 
100
  ### -------------------------------- ###
101
  ### rerun logic ###
 
95
  y_pred = model.predict(x_test)
96
 
97
  # save the model to file
98
+ with open('model.pkl, 'w+') as f:
99
+ pkl.dump(model, f)
100
 
101
  ### -------------------------------- ###
102
  ### rerun logic ###