Update app.py

app.py

@@ -20,10 +20,10 @@ import models_vit
 
 def prepare_model(chkpt_dir, arch='vit_large_patch14'):
     # build model
-    model = getattr(models_vit, arch)(global_pool=False)
+    model = getattr(models_vit, arch)(global_pool=True)
     # load model
     checkpoint = torch.load(chkpt_dir, map_location='cpu')
-    msg = model.load_state_dict(checkpoint['model'], strict=False)
+    msg = model.load_state_dict(checkpoint['model'], strict=True)
     print(msg)
     return model