gagan3012 commited on
Commit
87a1c03
·
2 Parent(s): 0432b0e 705340b

Merge branch 'master' of https://dagshub.com/gagan3012/summarization

Browse files
Files changed (1) hide show
  1. src/models/hf_upload.py +1 -0
src/models/hf_upload.py CHANGED
@@ -26,6 +26,7 @@ def upload(model_to_upload, model_name):
26
  """.strip()
27
 
28
  (Path(model_repo.local_dir) / "README.md").write_text(readme_txt)
 
29
  commit_url = model_repo.push_to_hub()
30
 
31
  print("Check out your model at:")
 
26
  """.strip()
27
 
28
  (Path(model_repo.local_dir) / "README.md").write_text(readme_txt)
29
+ model_to_upload.save_model(Path(model_repo.local_dir))
30
  commit_url = model_repo.push_to_hub()
31
 
32
  print("Check out your model at:")