Update README.md

README.md

@@ -10,4 +10,12 @@ license: apache-2.0
 library_name: keras
 ---
 
-# Video Vision Transformer on medmnist
+## Keras Implementation of Video Vision Transformer on medmnist
+
+This repo contains the model [to this Keras example on Video Vision Transformer](https://keras.io/examples/vision/vivit/).
+
+## Background Information 
+This example implements [ViViT: A Video Vision Transformer](https://arxiv.org/abs/2103.15691) by Arnab et al., a pure Transformer-based model for video classification. The authors propose a novel embedding scheme and a number of Transformer variants to model video clips.
+
+## Datasets
+We use the [MedMNIST v2: A Large-Scale Lightweight Benchmark for 2D and 3D Biomedical Image Classification](https://medmnist.com/) dataset.