ipd commited on
Commit
d359b22
·
1 Parent(s): 5f0e5c0

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ from torch.utils.mobile_optimizer import optimize_for_mobile
10
  model = timm.create_model('vit_base_patch16_224', pretrained=True)
11
  model.head = torch.nn.Linear(in_features=model.head.in_features, out_features=5)
12
 
13
- path = "opt_model.pt"
14
 
15
- model = model.jit.load(path)
16
 
17
  model.eval()
18
 
 
10
  model = timm.create_model('vit_base_patch16_224', pretrained=True)
11
  model.head = torch.nn.Linear(in_features=model.head.in_features, out_features=5)
12
 
13
+ #path = "opt_model.pt"
14
 
15
+ #model = model.jit.load(path)
16
 
17
  model.eval()
18