Update train.py

train.py

@@ -96,12 +96,23 @@ print('Test accuracy:', test_acc)
 # Show the model architecture
 model.summary()
 
-# Upload the model to your Hugging Face space repository
-push_to_hub_keras(
-    model, # model,  # Point to the SavedModel directory
-    repo_id="okeowo1014/kerascatanddog.keras",
-    commit_message="cats and dog image classifier with transfer learning",
-    tags=["image-classifier", "data-augmentation", "class-weights"],
-    include_optimizer=True,
-    token=sac,
-)
+# # Upload the model to your Hugging Face space repository
+# push_to_hub_keras(
+#     model, # model,  # Point to the SavedModel directory
+#     repo_id="okeowo1014/kerascatanddog.keras",
+#     commit_message="cats and dog image classifier with transfer learning",
+#     tags=["image-classifier", "data-augmentation", "class-weights"],
+#     include_optimizer=True,
+#     token=sac,
+# )
+# Save the model with a valid extension (recommended: .h5)
+model.save('cat_dog_classifierss.h5')
+
+# Authenticate with Hugging Face (replace YOUR_TOKEN with your actual token)
+from huggingface_hub import hf_hub_auth
+hf_hub_auth(token=sac)
+
+# Push the model to Hugging Face Hub with a descriptive name
+model_repo = push_to_hub_keras(model, repo_name="okeowo1014/kerascatanddog")  # Optional organization
+
+print(f"Model pushed to Hugging Face Hub: {model_repo}")