Spaces:
Running
Running
feat: add TorchSklearnWrapper import for model deserialization support
Browse files- local_demo.py +1 -0
local_demo.py
CHANGED
|
@@ -48,6 +48,7 @@ except ModuleNotFoundError as exc:
|
|
| 48 |
import numpy as np
|
| 49 |
|
| 50 |
from app.training.extract_features_batch import extract_sample_features
|
|
|
|
| 51 |
|
| 52 |
BASE_DIR = Path(__file__).resolve().parent
|
| 53 |
PROJECT_ROOT = BASE_DIR.parent
|
|
|
|
| 48 |
import numpy as np
|
| 49 |
|
| 50 |
from app.training.extract_features_batch import extract_sample_features
|
| 51 |
+
from app.training.train_deep_classifiers import TorchSklearnWrapper # noqa: F401 — required for pickle deserialization of DL models
|
| 52 |
|
| 53 |
BASE_DIR = Path(__file__).resolve().parent
|
| 54 |
PROJECT_ROOT = BASE_DIR.parent
|