diff --git a/src/litmodels/__init__.py b/src/litmodels/__init__.py
index e4f9786..479a80b 100644
--- a/src/litmodels/__init__.py
+++ b/src/litmodels/__init__.py
@@ -7,6 +7,6 @@
 _PACKAGE_ROOT = os.path.dirname(__file__)
 _PROJECT_ROOT = os.path.dirname(_PACKAGE_ROOT)
 
-from litmodels.io import download_model, load_model, save_model, upload_model
+from litmodels.io import download_model, load_model, save_model, upload_model, upload_model_files  # noqa: F401
 
 __all__ = ["download_model", "upload_model", "load_model", "save_model"]
diff --git a/src/litmodels/io/__init__.py b/src/litmodels/io/__init__.py
index 5f9a793..363f207 100644
--- a/src/litmodels/io/__init__.py
+++ b/src/litmodels/io/__init__.py
@@ -3,4 +3,4 @@
 from litmodels.io.cloud import download_model_files, upload_model_files  # noqa: F401
 from litmodels.io.gateway import download_model, load_model, save_model, upload_model
 
-__all__ = ["download_model", "upload_model", "load_model", "save_model"]
+__all__ = ["download_model", "upload_model", "upload_model_files", "load_model", "save_model"]