Update train.py

train.py

@@ -89,7 +89,7 @@ print('Test accuracy:', test_acc)
 # Export the model for Hugging Face Hub using tf.saved_model.save()
 export_dir = 'saved_model'  # Create a directory for the SavedModel
 tf.saved_model.save(model, export_dir)
-Save model weights to local directory
+# Save model weights to local directory
 # model.save_pretrained("kerascatanddog")
 
 # Upload the model to your Hugging Face space repository