Spaces:
Runtime error
Runtime error
Isabel Gwara commited on
Commit ·
b004973
1
Parent(s): f100c9a
Update app.py
Browse files
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 |
-
|
|
|
|
| 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 ###
|