Spaces:
Running
on
Zero
Running
on
Zero
Update core/process.py
Browse files- 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.")
|