Update app.py
Browse files
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(
|
| 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)
|