Transformers How to use klhung/ViLT_OCT with Transformers:
# Load model directly
from transformers import AutoProcessor, CustomViltForClassification
processor = AutoProcessor.from_pretrained("klhung/ViLT_OCT")
model = CustomViltForClassification.from_pretrained("klhung/ViLT_OCT") 
