DurgaDeepak commited on
Commit
2a02717
·
verified ·
1 Parent(s): 5ec100e

Update core/process.py

Browse files
Files changed (1) hide show
  1. core/process.py +1 -1
core/process.py CHANGED
@@ -77,7 +77,7 @@ def process_image(
77
  if run_det:
78
  logger.info(f"Running detection with model: {det_model}")
79
  load_start = time.time()
80
- model = get_model("detection", det_model, device="cuda" if torch.cuda.is_available() else "cpu")
81
  model.load_model()
82
  logger.info(f"{det_model} detection model loaded in {time.time() - load_start:.2f} seconds.")
83
  boxes = model.predict(image, conf_threshold=det_confidence)
 
77
  if run_det:
78
  logger.info(f"Running detection with model: {det_model}")
79
  load_start = time.time()
80
+ model = get_model("detection", det_model, device="cpu")
81
  model.load_model()
82
  logger.info(f"{det_model} detection model loaded in {time.time() - load_start:.2f} seconds.")
83
  boxes = model.predict(image, conf_threshold=det_confidence)