AILearning / README.md
Vincent Jiao
Upload cat or dog model
2644aef
metadata
title: Cat or Dog Classifier
emoji: 🐱🐢
colorFrom: yellow
colorTo: green
sdk: gradio
sdk_version: 5.49.1
app_file: app.py
pinned: false
license: apache-2.0

Cat or Dog Classifier 🐱🐢

A simple image classifier built with FastAI that distinguishes between cats and dogs.

Description

This application uses a pre-trained FastAI model to classify images as either cats or dogs. Simply upload an image, and the model will predict whether it contains a cat or a dog, along with confidence scores.

Features

  • πŸ–ΌοΈ Upload any cat or dog image
  • 🎯 Get instant predictions with confidence scores
  • πŸš€ Powered by FastAI and Gradio

Tech Stack

  • FastAI: Deep learning model training and inference
  • Gradio: User-friendly web interface
  • Hugging Face Spaces: Deployment platform

Local Development

To run this app locally:

# Install dependencies
pip install -r requirements.txt

# Run the app
python app.py

Model

The model is trained to classify images into two categories:

  • 🐢 Dog
  • 🐱 Cat