updated files

.gitignore

@@ -1,3 +1,3 @@
-/saved_model
+/saved_models
 .env
 /downloaded_models

push-model-to-huggingface-with-cli.sh

@@ -4,4 +4,4 @@
 # commit message is not customizable
 # https://huggingface.co/docs/huggingface_hub/guides/upload
 
-huggingface-cli upload manufy/mnist_model_keras saved_model . --commit-description "Uploaded folder using CLI" --token $HUGGINGFACE_TOKEN
+huggingface-cli upload manufy/mnist_model_keras saved_models . --commit-description "Uploaded folder using CLI" --token $HUGGINGFACE_TOKEN

push-model-to-huggingface-with-commit.py

@@ -22,15 +22,31 @@ load_dotenv(find_dotenv())
 token = os.getenv("HUGGINGFACE_TOKEN")
 formatted_now = datetime.now().strftime("%d/%m/%Y, %H:%M:%S")
 ic("--- Importing HF API ---")
-from huggingface_hub import HfApi, CommitOperationAdd, CommitOperationDelete
-ic("--- Executing commit with custom message ---")
-api = HfApi()
-operations = [
-    CommitOperationAdd(path_in_repo="saved_model", path_or_fileobj="mnist_model.keras")
-]
-api.create_commit(
-    repo_id="manufy/mnist_model_keras",
-    operations=operations,
-    commit_message=f"Updated model at  {formatted_now}.",
-    token=token
-)
+from huggingface_hub import HfApi, CommitOperationAdd, Repository
+
+# Crear una instancia del repositorio
+ic("--- Creating repository instance ---")
+repo = Repository(local_dir="saved_models")
+
+# Iterar sobre cada archivo en el directorio 'saved_models'
+for root, dirs, files in os.walk("saved_models"):
+    for file in files:
+        file_path = os.path.join(root, file)
+        # Agregar cada archivo al commit
+        repo.add(CommitOperationAdd(path_in_repo=file_path, path_or_fileobj=file_path))
+
+# Realizar el commit
+repo.commit("Commit message")
+
+#ic("--- Executing commit with custom message ---")
+#api = HfApi()∫
+#operations = [∫
+#    CommitOperationAdd(path_in_repo="saved_models", path_or_fileobj="saved_models")
+#    #CommitOperationAdd(path_in_repo="saved_models", path_or_fileobj="saved_models/keras/mnist_model.h5")
+#]
+#api.create_commit(
+##    repo_id="manufy/mnist_model_keras",
+#    operations=operations,
+#    commit_message=f"Updated models at  {formatted_now}.",
+#    token=token
+#)

push-model-to-huggingface-with-upload-folder.py

@@ -46,9 +46,9 @@ ic ("--- Importing HF API ---")
 from huggingface_hub import HfApi
 ic("--- Settig up HF API ---")
 HuggingFace_api = HfApi()
-ic("--- Uploading model ---")
+ic("--- Uploading models ---")
 HuggingFace_api.upload_folder(
-    folder_path="saved_model/",
+    folder_path="saved_models/",
     repo_id="manufy/mnist_model_keras",
     repo_type="model",
     token=token)

training.py

@@ -67,8 +67,8 @@ ic("------ Training model ------")
 # save model
 
 ic("------ Saving .h5 model ------")
-model.save('saved_model/keras/mnist_model.h5')
+model.save('saved_models/keras/mnist_model.h5')
 ic("------ Saving .keras model ------")
-model.save('saved_model/keras/mnist_model.keras')
+model.save('saved_models/keras/mnist_model.keras')
 ic("------ Exporting .keras model ------")
-model.export('saved_model/exported')
+model.export('saved_models/exported')