metadata
license: mit
tags:
- skin-disease
- medical
- computer-vision
- image-classification
datasets:
- ISIC
language:
- en
pipeline_tag: image-classification
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.