Hugging Face's logo

RationAI
/
vit-patch16-224-prostate

Transformers
PyTorch
Safetensors
Model card Files
xet
 Community
How to use from the
Use from the
Transformers library
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("RationAI/vit-patch16-224-prostate", dtype="auto")
Quick Links

ViT for prostate cancer classification 

model = ViTForImageClassification.from_pretrained("RationAI/vit-patch16-224-prostate")
processor = ViTImageProcessor.from_pretrained("RationAI/vit-patch16-224-prostate")
Downloads last month
221
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for RationAI/vit-patch16-224-prostate

Base model

google/vit-base-patch16-224
Finetuned
(2062)
this model