Update app.py
Browse files
app.py
CHANGED
|
@@ -190,7 +190,7 @@ class ViT(nn.Module):
|
|
| 190 |
|
| 191 |
|
| 192 |
# Load model weights
|
| 193 |
-
checkpoint = torch.load("
|
| 194 |
|
| 195 |
model = ViT(cfg).to(device)
|
| 196 |
|
|
|
|
| 190 |
|
| 191 |
|
| 192 |
# Load model weights
|
| 193 |
+
checkpoint = torch.load("best_CTD_ViT_CIFAR100_checkpoint.pth", map_location=device)
|
| 194 |
|
| 195 |
model = ViT(cfg).to(device)
|
| 196 |
|