okeowo1014 commited on
Commit
789498c
1 Parent(s): 9e50616

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -113,6 +113,6 @@ model.save('cat_dog_classifierss.h5')
113
  # hf_hub_auth(token=sac)
114
 
115
  # Push the model to Hugging Face Hub with a descriptive name
116
- model_repo = push_to_hub_keras(model, repo_name="okeowo1014/kerascatanddog",token=sac) # Optional organization
117
 
118
  print(f"Model pushed to Hugging Face Hub: {model_repo}")
 
113
  # hf_hub_auth(token=sac)
114
 
115
  # Push the model to Hugging Face Hub with a descriptive name
116
+ model_repo = push_to_hub_keras(model, repo_id="okeowo1014/kerascatanddog",token=sac) # Optional organization
117
 
118
  print(f"Model pushed to Hugging Face Hub: {model_repo}")