turhancan97 commited on
Commit
36a6b78
·
1 Parent(s): 8e1073d

bottom 25 model added to visualize in seperate tab

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ model_no_mode = torch.load(model_name, map_location='cpu')
18
  model_no_mode.eval()
19
  model_no_mode.to(device)
20
 
21
- model_name = "model/bottom_256/vit-t-mae-pretrain.pt"
22
  model_pca_mode = torch.load(model_name, map_location='cpu')
23
  model_pca_mode.eval()
24
  model_pca_mode.to(device)
 
18
  model_no_mode.eval()
19
  model_no_mode.to(device)
20
 
21
+ model_name = "model/bottom_25/vit-t-mae-pretrain.pt"
22
  model_pca_mode = torch.load(model_name, map_location='cpu')
23
  model_pca_mode.eval()
24
  model_pca_mode.to(device)