Update app.py

app.py

@@ -1,3 +1,18 @@
+import transformers
+import torch
+import torchvision
+from transformers import TrainingArguments, Trainer
+from transformers import ViTImageProcessor
+from transformers import ViTForImageClassification
+from torch.utils.data import DataLoader
+from datasets import load_dataset
+from torchvision.transforms import (CenterCrop, 
+                                    Compose, 
+                                    Normalize, 
+                                    RandomHorizontalFlip,
+                                    RandomResizedCrop, 
+                                    Resize, 
+                                    ToTensor)
 from transformers import ViTImageProcessor, ViTForImageClassification
 from PIL import Image
 import torch