metadata
library_name: transformers
tags: []
base_model:
- google/vit-base-patch16-224
ViT for prostate cancer classification
model = ViTForImageClassification.from_pretrained("RationAI/vit-patch16-224-prostate")
processor = ViTImageProcessor.from_pretrained("RationAI/vit-patch16-224-prostate")