polejowska commited on
Commit
00b292d
1 Parent(s): 6e3cc2f

Update constants.py

Browse files
Files changed (1) hide show
  1. constants.py +3 -2
constants.py CHANGED
@@ -10,8 +10,9 @@ COLORS = [
10
 
11
  MODELS_REPO = {
12
  "DETR-RESNET-50": "polejowska/detr-resnet-50-CD45RB-1000-att",
 
13
  "DETR-RESNET-101": "polejowska/detr-resnet-101-CD45RB-1000-att",
14
- "DETR-RESNET-50-TEST": "polejowska/detr-resnet50-leuk"
15
  }
16
 
17
- MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "DETR-RESNET-50-TEST"]
 
10
 
11
  MODELS_REPO = {
12
  "DETR-RESNET-50": "polejowska/detr-resnet-50-CD45RB-1000-att",
13
+ "DEFORMABLE-DETR": "polejowska/deformable-detr-resnet50-leuk",
14
  "DETR-RESNET-101": "polejowska/detr-resnet-101-CD45RB-1000-att",
15
+ "DETR-RESNET-50-TEST": "polejowska/detr-resnet50-leuk",
16
  }
17
 
18
+ MODELS_NAMES = ["DETR-RESNET-50", "DEFORMABLE-DETR-RESNET-50" "DETR-RESNET-101", "DETR-RESNET-50-TEST"]