Upload folder using huggingface_hub
Browse files- MLBaseModelDriver.py +1 -0
MLBaseModelDriver.py
CHANGED
|
@@ -55,6 +55,7 @@ class MLBaseModelDriver:
|
|
| 55 |
|
| 56 |
def __init__(self):
|
| 57 |
self.model, self.label_encoders, self.scaler = self.load_model()
|
|
|
|
| 58 |
self.preprocessor = DataPreprocessor(self.label_encoders, self.scaler)
|
| 59 |
|
| 60 |
def load_model(self) -> Tuple[any, any, any]:
|
|
|
|
| 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]:
|