Hugging Face's logo

Spaces:
cfoli
/
ChestVision
Sleeping

App Files Community
cfoli commited on
Commit
66210cd
·
1 Parent(s): 9d55218

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -47,12 +47,15 @@ configs = {
47
  }
48
 
49
  BACKBONE_REGISTRY = {
50
- "EfficientNet(b3)": {
51
- "torchvision_name": "efficientnet_b3",
52
- "ckpt": "EfficientNet(b3).ckpt"},
53
  "ConvNeXt(tiny)": {
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"},
 
47
  }
48
 
49
  BACKBONE_REGISTRY = {
50
+ "ConvNeXt(small)": {
51
+ "torchvision_name": "convnext_small",
52
+ "ckpt": "ConvNeXt(small).ckpt"},
53
  "ConvNeXt(tiny)": {
54
  "torchvision_name": "convnext_tiny",
55
  "ckpt": "ConvNeXt(tiny).ckpt"},
56
+ "EfficientNet(b3)": {
57
+ "torchvision_name": "efficientnet_b3",
58
+ "ckpt": "EfficientNet(b3).ckpt"},
59
  "EfficientNet(v2_small)": {
60
  "torchvision_name": "efficientnet_v2_s",
61
  "ckpt": "EfficientNet(v2_small).ckpt"},