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