can you also provide code for inference of VIT model
can you also provide code for inference of VIT model