push img clf models as private

huggingface · Feb 26, 2024 · 761c5f3 · 761c5f3
1 parent f46a4d0
commit 761c5f3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -30,4 +30,4 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="tensorflow")
 
 
-__version__ = "0.7.1.dev0"
+__version__ = "0.7.2.dev0"
diff --git a/src/autotrain/trainers/image_classification/__main__.py b/src/autotrain/trainers/image_classification/__main__.py
@@ -177,7 +177,7 @@ def train(config):
             save_training_params(config)
             logger.info("Pushing model to hub...")
             api = HfApi(token=config.token)
-            api.create_repo(repo_id=config.repo_id, repo_type="model")
+            api.create_repo(repo_id=config.repo_id, repo_type="model", private=True)
             api.upload_folder(folder_path=config.project_name, repo_id=config.repo_id, repo_type="model")
 
     if PartialState().process_index == 0:
Original file line number	Diff line number	Diff line change
Expand Up		@@ -30,4 +30,4 @@
		warnings.filterwarnings("ignore", category=UserWarning, module="tensorflow")


		__version__ = "0.7.1.dev0"
		__version__ = "0.7.2.dev0"