Update train.py

train.py

@@ -108,13 +108,21 @@ model.summary()
 # )
 # Save the model with a valid extension (recommended: .h5)
 model.save('cat_dog_classifierss.keras')
+
+push_to_hub_keras(
+    model,
+    repo_id="okeowo1014/catsanddogs",
+    commit_message="Initial commit",
+    token=sac,
+    include_optimizer=True,
+)
 # tf.keras.models.save_model(model, 'cat_dog_classifierss.keras')
 # Authenticate with Hugging Face (replace YOUR_TOKEN with your actual token)
 # from huggingface_hub import hf_hub_auth
 # hf_hub_auth(token=sac)
 # Recreate the exact same model purely from the file
-new_model = tf.keras.models.load_model('cat_dog_classifierss.keras')
+# new_model = tf.keras.models.load_model('cat_dog_classifierss.keras')
 # Push the model to Hugging Face Hub with a descriptive name
-model_repo = push_to_hub_keras(new_model, repo_id="okeowo1014/kerascatanddog",token=sac)  # Optional organization
+# model_repo = push_to_hub_keras(new_model, repo_id="okeowo1014/kerascatanddog",token=sac)  # Optional organization
 
 print(f"Model pushed to Hugging Face Hub: {model_repo}")