Update constants.py

constants.py

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