Update constants.py

constants.py

@@ -11,9 +11,8 @@ COLORS = [
 MODELS_REPO = {
     "DETR-RESNET-50": "polejowska/detr-resnet-50-CD45RB-1000-att",
     "DETR-RESNET-101": "polejowska/detr-resnet-101-CD45RB-1000-att",
-    "YOLOS-TINY": "polejowska/yolos-tiny-CD45RB-1000-att",
     "DETR-RESNET-50 (4)": "polejowska/detr-resnet-101-CD45RB-1000-att-e4",
-    "DETR-RESNET-50-NEW": "polejowska/detr-resnet50-cd45rb"
+    "DETR-RESNET-50-NEW": "polejowska/detr-resnet50-cd45rb-subdataset"
 }
 
-MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "YOLOS-TINY", "DETR-RESNET-50 (4)", "DETR-RESNET-50-NEW"]
+MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "DETR-RESNET-50 (4)", "DETR-RESNET-50-NEW"]