Skip to content

Commit e5d52d7

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent fc2c15a commit e5d52d7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

computer_vision/vision_transformer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import torch
1515
from transformers import ViTForImageClassification, ViTImageProcessor
1616

17+
1718
def vision_transformer_demo() -> None:
1819
"""
1920
Demonstrates Vision Transformer (ViT) on a sample image.
@@ -39,5 +40,6 @@ def vision_transformer_demo() -> None:
3940

4041
print(f"Predicted label: {predicted_label}")
4142

43+
4244
if __name__ == "__main__":
4345
vision_transformer_demo()

0 commit comments

Comments
 (0)