Update train.py

train.py

@@ -11,7 +11,7 @@ sac = os.getenv('accesstoken')
 
 
 # Define data paths (modify as needed)
-train_data_dir = 'train'
+train_data_dir = 'tt'
 validation_data_dir = 'valid'
 test_data_dir = 'valid'
 
@@ -88,12 +88,14 @@ print('Test accuracy:', test_acc)
 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
-upload_file(
-    model_path=export_dir,  # Point to the SavedModel directory
-    repo_id="okeowo1014/catsanddogs",
-    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
+# upload_file(
+#     model_path=export_dir,  # Point to the SavedModel directory
+#     repo_id="okeowo1014/catsanddogs",
+#     commit_message="cats and dog image classifier with transfer learning",
+#     tags=["image-classifier", "data-augmentation", "class-weights"],
+#     include_optimizer=True,
+#     token=sac
+# )
+# Upload the SavedModel directory to Hugging Face Hub
+upload_file(path_or_file=export_dir, repo_id="okeowo1014/catsanddogs", token=sac)