---
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")
```