You can upload your own image and see what the model says about it.
We can improve this model by changing the model architecture, and adding more training data. At this level I wanted to showcase my work towards implementation and deployment of Deep Learning model.
Model: "sequential_1" _________________________________________________________________ Layer (type) Output Shape Param # ================================================================= zero_padding2d_1 (ZeroPadd (None, 70, 70, 3) 0 ing2D) conv2d_1 (Conv2D) (None, 66, 66, 32) 2432 re_lu_1 (ReLU) (None, 66, 66, 32) 0 max_pooling2d_1 (MaxPoolin (None, 33, 33, 32) 0 g2D) dropout_1 (Dropout) (None, 33, 33, 32) 0 flatten_1 (Flatten) (None, 34848) 0 dense_1 (Dense) (None, 1) 34849 ================================================================= Total params: 37281 (145.63 KB) Trainable params: 37281 (145.63 KB) Non-trainable params: 0 (0.00 Byte) _________________________________________________________________