Add type hint and doctest to vision_transformer_demo

Author: dhruvidave348

computer_vision/vision_transformer.py

@@ -14,7 +14,6 @@
 import torch
 from transformers import ViTForImageClassification, ViTImageProcessor
 
-
 def vision_transformer_demo() -> None:
     """
     Demonstrates Vision Transformer (ViT) on a sample image.
@@ -40,6 +39,5 @@ def vision_transformer_demo() -> None:
 
     print(f"Predicted label: {predicted_label}")
 
-
 if __name__ == "__main__":
     vision_transformer_demo()