Upload folder using huggingface_hub

README.md

@@ -5,7 +5,7 @@ emoji: 🦀
 colorFrom: purple
 colorTo: purple
 sdk: gradio
-sdk_version: 3.1.7
+sdk_version: 3.50.2
 app_file: app.py
 pinned: false
 license: apache-2.0

steps/deploy_to_huggingface.py

@@ -15,11 +15,10 @@
 # limitations under the License.
 #
 
-import os
-from typing import Optional, List
+from typing_extensions import Annotated
 from huggingface_hub import create_branch, login, HfApi
 
-from zenml import step
+from zenml import step, log_artifact_metadata
 from zenml.client import Client
 from zenml.logger import get_logger
 
@@ -30,7 +29,7 @@ logger = get_logger(__name__)
 @step(enable_cache=False)
 def deploy_to_huggingface(
     repo_name: str,
-):
+) -> Annotated[str, "huggingface_url"]:
     """
     This step deploy the model to huggingface.
 
@@ -51,8 +50,20 @@ def deploy_to_huggingface(
             "this step will not work outside of a ZenML repo where the gradio folder is present."
         )
         raise
-    space = api.upload_folder(
+    url = api.upload_folder(
         folder_path=zenml_repo_root, repo_id=hf_repo.repo_id, repo_type="space", 
     )
-    logger.info(f"Space created: {space}")
+    repo_commits = api.list_repo_commits(
+        repo_id=hf_repo.repo_id,
+        repo_type="model",
+    )
+    log_artifact_metadata(
+        artifact_name="huggingface_url",
+        metadata={
+            "repo_id": hf_repo.repo_id,
+            "revision": repo_commits[0].commit_id,
+        },
+    )
+    logger.info(f"Model updated: {url}")
     ### YOUR CODE ENDS HERE ###
+    return url