Downloads · 30 days
8
2% of all-time downloads
real_classifier
nekofura/real_classifier
real_classifier is a image classification model from nekofura. Use it when you need a label for an image. It is set up for transformers.
This repository contains a pre-trained model for realistic image classification using the Vision Transformer (ViT) architecture, along with a Python script to perform inference on your own images. The model has been f…
All-time downloads
491
Public
Repo size
1.4 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin343 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- Model type
- vit
- Access
- Public
- Created
- Oct 22, 2023
- Updated
- Nov 2, 2023
- SHA
- fe242c47
Datasets
Model card
From the Hugging Face model README
Realistic Image Classification with Vits
This repository contains a pre-trained model for realistic image classification using the Vision Transformer (ViT) architecture, along with a Python script to perform inference on your own images. The model has been fine-tuned on a massive dataset of 20,000 high-quality images to deliver high-performance results, especially for Stable Diffusion XL (SDXL) tasks.
Hugging Face Model Hub
You can access and download the pre-trained model from the Hugging Face Model Hub using the following link: Real Classifier Model (Vits)
Requirements
To run the inference script, you need to have the following dependencies installed:
- PyTorch
- Transformers library by Hugging Face
- Pillow (PIL)
You can install these requirements using pip:
pip install torch transformers Pillow
Feel free to explore the capabilities of this model and contribute to its development by sharing feedback or improvements. If you have any questions or encounter any issues, please don't hesitate to open an issue in this repository.
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- Created
- Oct 22, 2023
- Updated
- Nov 2, 2023