DurgaDeepak commited on
Commit
9d9a80a
·
verified ·
1 Parent(s): 487b7f6

Update core/process.py

Browse files
Files changed (1) hide show
  1. core/process.py +1 -1
core/process.py CHANGED
@@ -76,9 +76,9 @@ def process_image(
76
  try:
77
  # Detection
78
  if run_det:
 
79
  logger.info(f"Running detection with model: {det_model}")
80
  load_start = time.time()
81
- #resolved_det_model = DETECTION_MODEL_MAP.get(det_model, det_model)
82
  model = get_model("detection", DETECTION_MODEL_MAP[det_model], device="cpu")
83
  model.load_model()
84
  logger.info(f"{det_model} detection model loaded in {time.time() - load_start:.2f} seconds.")
 
76
  try:
77
  # Detection
78
  if run_det:
79
+ if run_det:
80
  logger.info(f"Running detection with model: {det_model}")
81
  load_start = time.time()
 
82
  model = get_model("detection", DETECTION_MODEL_MAP[det_model], device="cpu")
83
  model.load_model()
84
  logger.info(f"{det_model} detection model loaded in {time.time() - load_start:.2f} seconds.")