From 387b4da8ff55f7e1e81ad80261bbd702c9b334a1 Mon Sep 17 00:00:00 2001 From: Seth Forsgren Date: Tue, 13 Dec 2022 17:07:49 -0800 Subject: [PATCH] updating mask image directory --- riffusion/baseten.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/riffusion/baseten.py b/riffusion/baseten.py index cb579db..db99dc8 100644 --- a/riffusion/baseten.py +++ b/riffusion/baseten.py @@ -133,7 +133,7 @@ class Model: # Load the mask image by ID if inputs.mask_image_id: - mask_image_path = Path(self._seed_images_dir, f"{inputs.mask_image_id}.png") + mask_image_path = Path(self._seed_images_dir, f"seed_images/{inputs.mask_image_id}.png") if not mask_image_path.is_file(): return f"Invalid mask image: {inputs.mask_image_id}", 400 mask_image = PIL.Image.open(str(mask_image_path)).convert("RGB")