fix(api): limit outpainting using image size params

ssube · Jan 14, 2023 · 34fa3f6 · 34fa3f6
1 parent 26a8ce7
commit 34fa3f6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -445,10 +445,10 @@ def inpaint():
     (model, provider, scheduler, prompt, negative_prompt, cfg, steps, height,
      width, seed) = pipeline_from_request()
 
-    left = get_and_clamp_int(request.args, 'left', 0, source_file.width)
-    right = get_and_clamp_int(request.args, 'right', 0, source_file.width)
-    top = get_and_clamp_int(request.args, 'top', 0, source_file.height)
-    bottom = get_and_clamp_int(request.args, 'bottom', 0, source_file.height)
+    left = get_and_clamp_int(request.args, 'left', 0, config_params.get('width').get('max'))
+    right = get_and_clamp_int(request.args, 'right', 0, config_params.get('width').get('max'))
+    top = get_and_clamp_int(request.args, 'top', 0, config_params.get('height').get('max'))
+    bottom = get_and_clamp_int(request.args, 'bottom', 0, config_params.get('height').get('max'))
 
     (output_file, output_full) = make_output_path(
         'inpaint', seed, (prompt, cfg, steps, height, width, seed, left, right, top, bottom))