Downloads · 30 days
25
3% of all-time downloads
skin-disease-classifier
HotJellyBean/skin-disease-classifier
skin-disease-classifier is a image classification model from HotJellyBean. Use it when you need a label for an image. The card lists the license as mit.
This model classifies skin lesions into 8 categories:
All-time downloads
828
Public
Parameters
86.8M
348 MB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.safetensors348 MB · 100%
Parameter types
How the weights are stored.
F3286.8M · 100%
Datasets
Spaces
Model card
From the Hugging Face model README
Skin Disease Classifier
This model classifies skin lesions into 8 categories:
- Actinic Keratosis
- Basal Cell Carcinoma
- Benign Keratosis
- Dermatofibroma
- Melanocytic Nevus
- Melanoma
- Squamous Cell Carcinoma
- Vascular Lesion
Model Details
- Architecture: Swin Transformer
- Dataset: ISIC (International Skin Imaging Collaboration)
- Classes: 8 skin disease categories
- Input: RGB images (224x224)
Usage
from transformers import AutoImageProcessor, AutoModelForImageClassification
from PIL import Image
processor = AutoImageProcessor.from_pretrained("HotJellyBean/skin-disease-classifier")
model = AutoModelForImageClassification.from_pretrained("HotJellyBean/skin-disease-classifier")
# Load and process image
image = Image.open("skin_lesion.jpg")
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
Disclaimer
This model is for educational and research purposes only. Always consult healthcare professionals for medical diagnosis.
- Task
- Image Classification
- Type
- swin
- License
- mit
- Languages
- en
- Created
- Dec 23, 2025
- Updated
- Dec 23, 2025