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

Add tags in modelcard

Browse files
Files changed (1) hide show
  1. src/model_trainer.py +1 -7
src/model_trainer.py CHANGED
@@ -89,13 +89,7 @@ def upload_model_to_hub(folder_path: Path, repo_name: str, token: str) -> str:
89
  )
90
  tags = info.tags
91
  tags.append("embeddinggemma-modkit")
92
-
93
- metadata_update(
94
- repo_id=repo_id,
95
- {"tags": tags},
96
- overwrite=True,
97
- token=token
98
- )
99
 
100
  return f"✅ Success! Model published at: {url}"
101
  except Exception as e:
 
89
  )
90
  tags = info.tags
91
  tags.append("embeddinggemma-modkit")
92
+ metadata_update(repo_id, {"tags": tags}, overwrite=True, token=token)
 
 
 
 
 
 
93
 
94
  return f"✅ Success! Model published at: {url}"
95
  except Exception as e: