bebechien commited on
Commit
ad4164f
·
verified ·
1 Parent(s): e7a7086

Add tags in modelcard

Browse files
Files changed (1) hide show
  1. 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