Hugging Face's logo

Spaces:
amanm10000
/
Simple-Resnet-50-Classifier
Sleeping

App Files Community
Simple-Resnet-50-Classifier / README.md
amanm10000's picture
amanm10000
Modify README.md, app.py
2231a45
preview code
|
raw
history blame contribute delete
1.09 kB

A newer version of the Streamlit SDK is available: 1.54.0

Upgrade
metadata 
title: Simple Resnet 50 Classifier
emoji: 🏆
colorFrom: green
colorTo: yellow
sdk: streamlit
sdk_version: 1.40.0
app_file: app.py
pinned: false
license: mit

Simple ResNet-50 Classifier 🏆

A Streamlit web application that uses the pre-trained ResNet-50 model for image classification.

Features

  • Image upload through drag-and-drop or file browser
  • Real-time image classification
  • Top 5 predictions with confidence scores
  • Support for JPG, JPEG, and PNG formats
  • User-friendly interface
  • Built on PyTorch and Streamlit

Technical Details

  • Framework: Streamlit v1.40.0
  • Model: Pre-trained ResNet-50
  • Image Processing: PIL and torchvision
  • Classification: 1000 ImageNet classes

Requirements

The project requires the following dependencies:

streamlit==1.24.0
torch==2.0.1
torchvision==0.15.2
Pillow==9.5.0

Usage

  1. Clone the repository
  2. Install dependencies:
    pip install -r requirements.txt
  3. Run the app:
    streamlit run app.py

License

This project is licensed under the MIT License.