Hugging Face's logo

Spaces:
turhancan97
/
MAE-Demo
Running

App Files Community
turhancan97 commited on
Commit
ee8e678
•
1 Parent(s): a6f67a5

new images added

Browse files
Files changed (4) hide show
  1. app.py +1 -1
  2. images/airplane.jpg +0 -0
  3. images/horse.jpg +0 -0
  4. images/truck.jpg +0 -0
app.py CHANGED
@@ -10,7 +10,7 @@ from torchvision.transforms import v2
10
 
11
  from model import MAE_ViT, MAE_Encoder, MAE_Decoder, MAE_Encoder_FeatureExtractor
12
 
13
- path = [['images/cat.jpg'], ['images/dog.jpg']]
14
  model_name = "vit-t-mae-pretrain.pt"
15
  model = torch.load(model_name, map_location='cpu')
16
 
 
10
 
11
  from model import MAE_ViT, MAE_Encoder, MAE_Decoder, MAE_Encoder_FeatureExtractor
12
 
13
+ path = [['images/cat.jpg'], ['images/dog.jpg'], ['images/horse.jpg'], ['images/airplane.jpg'], ['images/truck.jpg']]
14
  model_name = "vit-t-mae-pretrain.pt"
15
  model = torch.load(model_name, map_location='cpu')
16
 
images/airplane.jpg ADDED
images/horse.jpg ADDED
images/truck.jpg ADDED