diff --git a/predict.py b/predict.py index 045193f..8a384c4 100644 --- a/predict.py +++ b/predict.py @@ -311,7 +311,7 @@ def process_relight( bg_source, ) results = [(x * 255.0).clip(0, 255).astype(np.uint8) for x in results] - return results + extra_images + return results # + extra_images @torch.inference_mode() @@ -839,4 +839,4 @@ def predict( Image.fromarray(img).save(img_path, **save_params) output_paths.append(Path(img_path)) - return output_paths + return output_paths[::-1]