Mirrors
/
riffusion-inference
mirror of https://github.com/riffusion/riffusion.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updating mask image directory

This commit is contained in:
Seth Forsgren 2022-12-13 17:07:49 -08:00
parent 2fea921515
commit 387b4da8ff
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
riffusion/baseten.py
View File

@ -133,7 +133,7 @@ class Model:
# Load the mask image by ID # Load the mask image by ID
if inputs.mask_image_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(): if not mask_image_path.is_file():
return f"Invalid mask image: {inputs.mask_image_id}", 400 return f"Invalid mask image: {inputs.mask_image_id}", 400
mask_image = PIL.Image.open(str(mask_image_path)).convert("RGB") mask_image = PIL.Image.open(str(mask_image_path)).convert("RGB")