build: improved model tagging script

huspacy · Oct 27, 2023 · 860cce9 · 860cce9
1 parent 6cd3590
commit 860cce9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/scripts/tag_model.py b/scripts/tag_model.py
@@ -8,10 +8,14 @@
 
 
 @app.command()
-def main(repo_name: str, tag: str, message: Optional[str] = typer.Option(None, "-m", help="Tag message"),
-         organization=typer.Option("huspacy", "-o", help="Organization slug")):
+def main(repo_name: str, tag: str,
+         message: Optional[str] = typer.Option(None, "-m", help="Tag message"),
+         organization: str=typer.Option("huspacy", "-o", help="Organization slug"),
+         delete_existing: bool = typer.Option(False, "-d", help="Delete tag if exists")):
     # token = typer.prompt("Auth token")
     api = HfApi()
+    if delete_existing:
+        api.delete_tag(repo_id=f"{organization}/{repo_name}", tag=tag)
     api.create_tag(repo_id=f"{organization}/{repo_name}", tag=tag, tag_message=message)