vision-classifier / README.md
spML's picture
Update README.md
4720468 verified

A newer version of the Gradio SDK is available: 5.49.1

Upgrade
metadata
title: Simple Vision Classifier (ViT)
emoji: 馃惣
colorFrom: green
colorTo: blue
sdk: gradio
app_file: app.py
pinned: false
sdk_version: 5.47.2

馃惣 Simple Image Classifier (ViT)

Este Space demuestra un clasificador de im谩genes con modelos del Hub (ViT/DeiT/ResNet) usando transformers + gradio.

馃殌 C贸mo usarlo (local)

pip install -r requirements.txt
python app.py

馃敡 Cambiar el modelo

En la UI hay un desplegable con modelos populares:

  • google/vit-base-patch16-224
  • facebook/deit-base-patch16-224
  • microsoft/resnet-50

Puedes a帽adir m谩s en MODEL_OPTIONS dentro de app.py.