Upload folder using huggingface_hub

MLBaseModelDriver.py

@@ -53,8 +53,8 @@ class ProcessedSynapse(TypedDict):
 # --------- Класс MLBaseModelDriver ---------
 class MLBaseModelDriver:
 
-    def __init__(self):
-        self.model_path = model_path  # сначала сохраняем путь
+    def __init__(self, model_path: str):
+        self.model_path = model_path  # сохраняем путь модели
         self.model, self.label_encoders, self.scaler = self.load_model()
         self.preprocessor = DataPreprocessor(self.label_encoders, self.scaler)
 
@@ -64,7 +64,7 @@ class MLBaseModelDriver:
         model_class = self._import_model_class(model_class_file)
 
         model = model_class(input_dim=4)
-        state_dict = torch.load(model_file, weights_only=False)
+        # Здесь загружаем сохранённые веса модели
         checkpoint = torch.load(self.model_path, map_location=torch.device("cpu"))
         model.load_state_dict(checkpoint['model_state_dict'])
         self.preprocessor = checkpoint.get('preprocessor', None)