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