polejowska commited on
Commit
9f3ba68
1 Parent(s): 7f7eaee

Update constants.py

Browse files
Files changed (1) hide show
  1. constants.py +4 -3
constants.py CHANGED
@@ -9,9 +9,10 @@ COLORS = [
9
  ]
10
 
11
  MODELS_REPO = {
12
- "DETR-RESENT-50": "polejowska/detr-resnet-50-CD45RB-1000-att",
13
- "DETR-RESENT-101": "polejowska/detr-resnet-101-CD45RB-1000-att",
14
  "YOLOS-TINY": "polejowska/yolos-tiny-CD45RB-1000-att",
 
15
  }
16
 
17
- MODELS_NAMES = ["DETR-RESENT-50", "DETR-RESENT-101", "YOLOS-TINY"]
 
9
  ]
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
  "YOLOS-TINY": "polejowska/yolos-tiny-CD45RB-1000-att",
15
+ "DETR-RESNET-50 (4)": "polejowska/detr-resnet-101-CD45RB-1000-att-e4",
16
  }
17
 
18
+ MODELS_NAMES = ["DETR-RESENT-50", "DETR-RESENT-101", "YOLOS-TINY", "DETR-RESNET-50 (4)"]