fix(api): pass image size to upscale job

ssube · Jan 18, 2023 · cf6a151 · cf6a151
1 parent dc84bec
commit cf6a151
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -455,7 +455,7 @@ def upscale():
 
     source_image.thumbnail((size.width, size.height))
     executor.submit_stored(output, run_upscale_pipeline,
-                           context, params, output, upscale, source_image)
+                           context, params, size, output, upscale, source_image)
 
     return jsonify({
         'output': output,