cfoli commited on
Commit
5db8e46
·
1 Parent(s): 22e4bb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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