Update core/process.py

core/process.py

@@ -78,8 +78,8 @@ def process_image(
         if run_det:
             logger.info(f"Running detection with model: {det_model}")
             load_start = time.time()
-            resolved_det_model = DETECTION_MODEL_MAP.get(det_model, det_model)
-            model = get_model("detection", resolved_det_model, device="cpu")
+            #resolved_det_model = DETECTION_MODEL_MAP.get(det_model, det_model)
+            model = get_model("detection", DETECTION_MODEL_MAP[det_model], device="cpu")
             model = model.load_model()
             logger.info(f"{det_model} detection model loaded in {time.time() - load_start:.2f} seconds.")
             boxes = model.predict(image, conf_threshold=det_confidence)