Ahmed766/computer-vision-demo

Computer Vision Model

This model performs image classification using a pre-trained ResNet model.

Features

• —Image classification across 1000 object categories
• —Preprocessing included
• —Batch processing support
• —Configurable confidence threshold

Usage

from image_classifier import ImageClassifier

classifier = ImageClassifier()
result = classifier.classify_image("path/to/image.jpg")
print(result)

Model Details

• —Architecture: ResNet-50
• —Dataset: ImageNet
• —Top-1 Accuracy: 76.2%

Hugging Face Space

This model can be deployed as a Hugging Face Space with a Gradio interface for easy interaction.

Installation

pip install -r requirements.txt

Local Testing

python test_model.py