Upload folder using huggingface_hub

MLBaseModelDriver.py

@@ -64,7 +64,7 @@ class MLBaseModelDriver:
 
         model = model_class(input_dim=4)
         state_dict = torch.load(model_file, weights_only=False)
-        checkpoint = torch.load(model_path, map_location=torch.device("cpu"))
+        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)
         self.input_dim = checkpoint.get('input_dim', None)