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 = []