Update app.py

app.py

@@ -5,7 +5,7 @@ import transformers
 from transformers import ViTImageProcessor
 
 # For Model
-from transformers import ViTModel, ViTConfig
+from transformers import ViTModel, ViTConfig, pipeline
 
 # For data augmentation
 from torchvision import transforms, datasets