okeowo1014 commited on
Commit
300365a
1 Parent(s): 4fac414

Update train.py

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -112,7 +112,7 @@ model.save('cat_dog_classifierss.keras','',save_format='tf',)
112
  # from huggingface_hub import hf_hub_auth
113
  # hf_hub_auth(token=sac)
114
  # Recreate the exact same model purely from the file
115
- new_model = keras.models.load_model('cat_dog_classifierss.keras')
116
  # Push the model to Hugging Face Hub with a descriptive name
117
  model_repo = push_to_hub_keras(new_model, repo_id="okeowo1014/kerascatanddog",token=sac) # Optional organization
118
 
 
112
  # from huggingface_hub import hf_hub_auth
113
  # hf_hub_auth(token=sac)
114
  # Recreate the exact same model purely from the file
115
+ new_model = tf.keras.models.load_model('cat_dog_classifierss.keras')
116
  # Push the model to Hugging Face Hub with a descriptive name
117
  model_repo = push_to_hub_keras(new_model, repo_id="okeowo1014/kerascatanddog",token=sac) # Optional organization
118