fix(api): use correct base path for debug images

ssube · Jan 17, 2023 · 634d2e8 · 634d2e8
1 parent 5ba752e
commit 634d2e8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/onnx_web/pipeline.py b/api/onnx_web/pipeline.py
@@ -166,9 +166,9 @@ def run_inpaint_pipeline(
         mask_filter=mask_filter)
 
     if environ.get('DEBUG') is not None:
-        source_image.save(safer_join(output.path, 'last-source.png'))
-        mask_image.save(safer_join(output.path, 'last-mask.png'))
-        noise_image.save(safer_join(output.path, 'last-noise.png'))
+        source_image.save(safer_join(ctx.output_path, 'last-source.png'))
+        mask_image.save(safer_join(ctx.output_path, 'last-mask.png'))
+        noise_image.save(safer_join(ctx.output_path, 'last-noise.png'))
 
     image = pipe(
         params.prompt,