Update app.py

app.py

@@ -156,7 +156,7 @@ class ViT(nn.Module):
 # ------------------------
 # Load model weights
 model = ViT(cfg).to(device)
-model.load_state_dict(torch.load("", map_location=device))
+model.load_state_dict(torch.load("best_vit_cifar10 (1).pt", map_location=device))
 model.eval()
 
 # ------------------------