Spaces:
Sleeping
Sleeping
Update backend/utils/model_io.py
Browse files- backend/utils/model_io.py +10 -12
backend/utils/model_io.py
CHANGED
|
@@ -1,12 +1,10 @@
|
|
| 1 |
-
import
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import joblib
|
| 3 |
+
|
| 4 |
+
class ModelIO:
|
| 5 |
+
def save(self, model, path):
|
| 6 |
+
os.makedirs(os.path.dirname(path), exist_ok=True)
|
| 7 |
+
joblib.dump(model, path)
|
| 8 |
+
|
| 9 |
+
def load(self, path):
|
| 10 |
+
return joblib.load(path)
|
|
|
|
|
|