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
+# from huggingface_hub import push_to_hub
+from huggingface_hub import HfApi
+api = HfApi()
 
 # Environment variable for Hugging Face token
 sac = os.getenv('accesstoken')
@@ -88,7 +90,7 @@ export_dir = 'saved_model'  # Create a directory for the SavedModel
 tf.saved_model.save(model, export_dir)
 
 # Upload the model to your Hugging Face space repository
-push_to_hub(
+api.push_to_hub(
     model_path=export_dir,  # Point to the SavedModel directory
     repo_id="okeowo1014/catsanddogs",
     commit_message="cats and dog image classifier with transfer learning",