π± Cat vs Dog Image Classifier
A TensorFlow CNN model for binary image classification.
Classes
- Cat
- Dog
Architecture
- Conv2D
- MaxPooling2D
- Dense
- Dropout
Input Size
150x150 RGB
Accuracy
Validation Accuracy: 95%
Installation
pip install tensorflow pillow numpy
Download
from huggingface_hub import hf_hub_download
model_path = hf_hub_download(
repo_id="PatelDhvanit/Cat-Dog-Classifier",
filename="cat_dog_model.h5"
)
Prediction
import tensorflow as tf
model=tf.keras.models.load_model(model_path)
Author
Dhvanit Patel
Inference Providers NEW
This model isn't deployed by any Inference Provider. π Ask for provider support