Update constants.py

constants.py

@@ -13,6 +13,7 @@ MODELS_REPO = {
     "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"
 }
 
-MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "YOLOS-TINY", "DETR-RESNET-50 (4)"]
+MODELS_NAMES = ["DETR-RESNET-50", "DETR-RESNET-101", "YOLOS-TINY", "DETR-RESNET-50 (4)", "DETR-RESNET-50-NEW"]