From a04471d5745f3fd032b2c1d96bed02ae28990ae5 Mon Sep 17 00:00:00 2001 From: Fam Zheng Date: Sat, 27 Dec 2025 12:38:02 +0000 Subject: [PATCH] common.py: fix predict --- emblem5/ai/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/emblem5/ai/common.py b/emblem5/ai/common.py index 3d4c036..b4484b7 100644 --- a/emblem5/ai/common.py +++ b/emblem5/ai/common.py @@ -517,7 +517,7 @@ def verify_frame(model, transforms, frame_img, orig_img): side_by_side_img = side_by_side_img.convert('RGB') with tempfile.NamedTemporaryFile(suffix='.jpg') as f: side_by_side_img.save(f.name) - return predict(model, transforms, Image.open(f.name).convert('RGB')) + return predict(model, transforms, Image.open(f.name).convert('RGB'), ncells=3) def parse_ranges(s): ret = []