okeowo1014 commited on
Commit
dc92021
1 Parent(s): 3561ca3

Update train3.py

Browse files
Files changed (1) hide show
  1. train3.py +1 -1
train3.py CHANGED
@@ -75,4 +75,4 @@ test_loss, test_acc = model.evaluate(test_generator)
75
  print('Test accuracy:', test_acc)
76
 
77
  # Save the model for future use (optional)
78
- model.save('cat_dog_classifier.h5')
 
75
  print('Test accuracy:', test_acc)
76
 
77
  # Save the model for future use (optional)
78
+ model.save('cat_dog_classifier.keras')