Upload folder using huggingface_hub
Browse files- MLBaseModelDriver.py +1 -1
MLBaseModelDriver.py
CHANGED
|
@@ -54,8 +54,8 @@ class ProcessedSynapse(TypedDict):
|
|
| 54 |
class MLBaseModelDriver:
|
| 55 |
|
| 56 |
def __init__(self):
|
|
|
|
| 57 |
self.model, self.label_encoders, self.scaler = self.load_model()
|
| 58 |
-
self.model_path = model_path # ← добавь эту строку
|
| 59 |
self.preprocessor = DataPreprocessor(self.label_encoders, self.scaler)
|
| 60 |
|
| 61 |
def load_model(self) -> Tuple[any, any, any]:
|
|
|
|
| 54 |
class MLBaseModelDriver:
|
| 55 |
|
| 56 |
def __init__(self):
|
| 57 |
+
self.model_path = model_path # сначала сохраняем путь
|
| 58 |
self.model, self.label_encoders, self.scaler = self.load_model()
|
|
|
|
| 59 |
self.preprocessor = DataPreprocessor(self.label_encoders, self.scaler)
|
| 60 |
|
| 61 |
def load_model(self) -> Tuple[any, any, any]:
|