Gagan Bhatia commited on
Commit
c201811
1 Parent(s): c0bd5ec

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -561,3 +561,6 @@ class Summarization:
561
  del hf_password
562
  model_url = HfApi().create_repo(token=token, name=model_name, exist_ok=True)
563
  model_repo = Repository(
 
 
 
 
561
  del hf_password
562
  model_url = HfApi().create_repo(token=token, name=model_name, exist_ok=True)
563
  model_repo = Repository(
564
+ "./model",
565
+ clone_from=model_url,
566
+ use_auth_token=token,