Update README.md

README.md

@@ -22,6 +22,7 @@ model_test.eval()  # Set the model to evaluation mode
 
 The model implementation is here:
 ```
+from transformers import ViTModel
 class MultiModalModel(nn.Module):
     def __init__(self, num_classes=2):
         super(MultiModalModel, self).__init__()