Spaces:
Runtime error
Runtime error
polejowska
commited on
Commit
•
9f3ba68
1
Parent(s):
7f7eaee
Update constants.py
Browse files- constants.py +4 -3
constants.py
CHANGED
@@ -9,9 +9,10 @@ COLORS = [
|
|
9 |
]
|
10 |
|
11 |
MODELS_REPO = {
|
12 |
-
"DETR-
|
13 |
-
"DETR-
|
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)"]
|