Spaces:
Runtime error
Runtime error
Isabel Gwara
commited on
Commit
·
5f22042
1
Parent(s):
a7c03e5
Update app.py
Browse files
app.py
CHANGED
|
@@ -95,7 +95,7 @@ def train_model():
|
|
| 95 |
y_pred = model.predict(x_test)
|
| 96 |
|
| 97 |
# save the model to file
|
| 98 |
-
with open('model.pkl', '
|
| 99 |
pkl.dump(model, f)
|
| 100 |
|
| 101 |
### -------------------------------- ###
|
|
|
|
| 95 |
y_pred = model.predict(x_test)
|
| 96 |
|
| 97 |
# save the model to file
|
| 98 |
+
with open('model.pkl', 'wb') as f:
|
| 99 |
pkl.dump(model, f)
|
| 100 |
|
| 101 |
### -------------------------------- ###
|