Spaces:
Sleeping
Sleeping
Commit
·
36a6b78
1
Parent(s):
8e1073d
bottom 25 model added to visualize in seperate tab
Browse files
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/
|
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)
|