[EAGLE-3758] remove model (#167)

* remove model * remove more models
Clarifai · Dec 14, 2023 · 48bfd32 · 48bfd32
1 parent bcc6e36
commit 48bfd32
Showing 1 changed file with 1 addition and 26 deletions.
diff --git a/tests/public_models/public_test_helper.py b/tests/public_models/public_test_helper.py
@@ -55,31 +55,6 @@
 # general visual detection models (yolo, detic)
 # Data Structure: {MODEL_NAME: [title, model, app, user]}
 OBJECT_DETECTION_MODELS = {
-    "YOLOV6_S": [
-        "general-detector-yolov6s-coco",
-        "yolov6s-coco",
-        "yolov6",
-        "meituan",
-    ],
-    "YOLOV6_NANO": [
-        "general-detector-yolov6n-coco",
-        "yolov6n-coco",
-        "yolov6",
-        "meituan",
-    ],
-    "YOLOV7": ["general-detector-yolov7-coco", "yolov7", "yolov7", "wongkinyiu"],
-    "YOLOV7_E6": [
-        "general-image-detector-yolov7-e6-coco",
-        "yolov7-e6",
-        "yolov7",
-        "wongkinyiu",
-    ],
-    "YOLOV7_X": [
-        "general-image-detector-yolov7-x-coco",
-        "yolov7-x",
-        "yolov7",
-        "wongkinyiu",
-    ],
     "BLAZE_FACE_DETECTOR": [
         "general-image-detector-blazeface_ssh-widerface",
         "general-image-detector-blazeface_ssh-widerface",
@@ -111,7 +86,7 @@
         "facebook",
     ],
 }
-SHORT_OBJECT_DETECTION_MODEL_KEYS = ["YOLOV6_S", "YOLOV7", "DETIC_CLIP_R50"]
+SHORT_OBJECT_DETECTION_MODEL_KEYS = ["DETIC_CLIP_R50"]
 OBJECT_DETECTION_MODELS_SHORT = {
     k: OBJECT_DETECTION_MODELS[k] for k in SHORT_OBJECT_DETECTION_MODEL_KEYS
 }