Add tags in modelcard
Browse files- src/model_trainer.py +1 -1
src/model_trainer.py
CHANGED
|
@@ -87,7 +87,7 @@ def upload_model_to_hub(folder_path: Path, repo_name: str, token: str) -> str:
|
|
| 87 |
repo_id=repo_id,
|
| 88 |
token=token
|
| 89 |
)
|
| 90 |
-
tags = info.tags
|
| 91 |
tags.append("embeddinggemma-modkit")
|
| 92 |
metadata_update(repo_id, {"tags": tags}, overwrite=True, token=token)
|
| 93 |
|
|
|
|
| 87 |
repo_id=repo_id,
|
| 88 |
token=token
|
| 89 |
)
|
| 90 |
+
tags = info.card_data.tags
|
| 91 |
tags.append("embeddinggemma-modkit")
|
| 92 |
metadata_update(repo_id, {"tags": tags}, overwrite=True, token=token)
|
| 93 |
|