polejowska commited on
Commit
710ed30
1 Parent(s): 27e0297

Update constants.py

Browse files
Files changed (1) hide show
  1. constants.py +2 -1
constants.py CHANGED
@@ -11,8 +11,9 @@ COLORS = [
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"]
 
11
  MODELS_REPO = {
12
  "DETR-RESNET-50": "polejowska/detr-resnet-50-CD45RB-1000-att",
13
  "DEFORMABLE-DETR": "polejowska/deformable-detr-resnet50-leuk",
14
+ "CONDITIONAL-DETR": "polejowska/cdetr-cd45rb-s",
15
  "DETR-RESNET-101": "polejowska/detr-resnet-101-CD45RB-1000-att",
16
  "DETR-RESNET-50-TEST": "polejowska/detr-resnet50-leuk",
17
  }
18
 
19
+ MODELS_NAMES = ["DETR-RESNET-50", "DEFORMABLE-DETR", "CONDITIONAL-DETR", "DETR-RESNET-101", "DETR-RESNET-50-TEST"]