Update app.py

app.py

@@ -14,6 +14,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load ViT model and processor from Hugging Face
 model_path = "andromeda01111/ViT_BCC"
 
+vit_processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
 vit_model = ViTForImageClassification.from_pretrained(model_path).to(device)
 vit_model.eval()
 vit_processor = ViTImageProcessor.from_pretrained(model_path)