Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,13 +54,13 @@ BACKBONE_REGISTRY = {
|
|
| 54 |
"torchvision_name": "convnext_tiny",
|
| 55 |
"ckpt": "ConvNeXt(tiny).ckpt"},
|
| 56 |
"EfficientNet(v2_small)": {
|
| 57 |
-
"torchvision_name": "efficientnet_v2_s
|
| 58 |
"ckpt": "EfficientNet(v2_small).ckpt"},
|
| 59 |
"RegNet(x3_2GF)": {
|
| 60 |
-
"torchvision_name": "regnet_x_3_2gf
|
| 61 |
"ckpt": "RegNet(x3_2GF).ckpt"},
|
| 62 |
"ResNet50": {
|
| 63 |
-
"torchvision_name": "resnet50
|
| 64 |
"ckpt": "ResNet50.ckpt"}
|
| 65 |
}
|
| 66 |
|
|
|
|
| 54 |
"torchvision_name": "convnext_tiny",
|
| 55 |
"ckpt": "ConvNeXt(tiny).ckpt"},
|
| 56 |
"EfficientNet(v2_small)": {
|
| 57 |
+
"torchvision_name": "efficientnet_v2_s",
|
| 58 |
"ckpt": "EfficientNet(v2_small).ckpt"},
|
| 59 |
"RegNet(x3_2GF)": {
|
| 60 |
+
"torchvision_name": "regnet_x_3_2gf",
|
| 61 |
"ckpt": "RegNet(x3_2GF).ckpt"},
|
| 62 |
"ResNet50": {
|
| 63 |
+
"torchvision_name": "resnet50",
|
| 64 |
"ckpt": "ResNet50.ckpt"}
|
| 65 |
}
|
| 66 |
|