ImageClassification / README.md
DrDavis's picture
Update README.md
d298ec1 verified
|
raw
history blame
694 Bytes
metadata
title: Image Classification
sdk: gradio
emoji: 👁
colorFrom: green
colorTo: purple

Image Classification — Gradio + Hugging Face

This Space wraps a pretrained image classifier (Vision Transformer) with a simple Gradio UI.

How to use

  1. Upload a photo (PNG/JPG).
  2. Choose how many classes to display (Top‑K).
  3. Click Classify to see the predicted labels with confidence scores.

Model

  • Default: google/vit-base-patch16-224 (ImageNet‑1k)
  • Alternatives that work on CPU Basic:
    • microsoft/resnet-50
    • facebook/convnext-tiny-224

Notes: The classifier is trained on ImageNet‑1k; some objects (logos, memes, niche items) may not map cleanly to labels.