HotJellyBean/skin-disease-classifier

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.