Update train.py

train.py

@@ -3,7 +3,9 @@ import tensorflow as tf
 from tensorflow.keras.preprocessing.image import ImageDataGenerator
 from tensorflow.keras.applications import VGG16
 from tensorflow.keras.layers import Flatten, Dense
-from huggingface_hub import push_to_hub_keras
+# from huggingface_hub import push_to_hub_keras
+from huggingface_hub import hf_hub_url, push_to_hub_keras
+
 # from huggingface_hub import upload_file
 from huggingface_hub import HfApi
 api = HfApi()
@@ -94,8 +96,7 @@ tf.saved_model.save(model, export_dir)
 
 # Upload the model to your Hugging Face space repository
 push_to_hub_keras(
-    model=export_dir,
-    # model,  # Point to the SavedModel directory
+    model, # model,  # Point to the SavedModel directory
     repo_id="okeowo1014/kerascatanddog",
     commit_message="cats and dog image classifier with transfer learning",
     tags=["image-classifier", "data-augmentation", "class-weights"],