offry commited on
Commit
4bc48f4
·
verified ·
1 Parent(s): f84d569

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def knn_calc(image_name, query_feature, features):
53
  return -dist.item()
54
 
55
  checkpoint_path = "multi_label.pth.tar"
56
- resnet = models.resnet101(pretrained=False)
57
  num_ftrs = resnet.fc.in_features
58
  resnet.fc = nn.Linear(num_ftrs, 13)
59
  model = Resnet_with_skip(resnet)
 
53
  return -dist.item()
54
 
55
  checkpoint_path = "multi_label.pth.tar"
56
+ resnet = models.resnet101()
57
  num_ftrs = resnet.fc.in_features
58
  resnet.fc = nn.Linear(num_ftrs, 13)
59
  model = Resnet_with_skip(resnet)