From cf5b116284416dd370a2272cd35015b7ac8bc4de Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Mon, 4 Jul 2022 18:41:00 -0400 Subject: [PATCH] update predict.py --- replicate/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/replicate/predict.py b/replicate/predict.py index 5304d16..684274a 100644 --- a/replicate/predict.py +++ b/replicate/predict.py @@ -32,7 +32,7 @@ class Predictor(BasePredictor): ), ) -> Path: - def handle_intermediate_image(i: int, image: Image.Image) -> Path: + def handle_intermediate_image(i: int, image: Image.Image): if i + 1 == 16: return out_path = Path(tempfile.mkdtemp()) / 'output.jpg' image.save(str(out_path))