Spaces:
Runtime error
Runtime error
Merge branch 'master' of https://dagshub.com/gagan3012/summarization
Browse files- 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:")
|