Upload app.py

app.py

@@ -1222,9 +1222,9 @@ with gr.Blocks() as demo:
 
     run_btn.click(
         fn=infer,
-        inputs=[hidden_images_b64, prompt, seed, randomize_seed, guidance_scale, steps],
+        inputs=[hidden_images_b64, prompt, seed, randomize_seed, guidance_scale, steps, aspect_ratio, output_count],
         outputs=[result_data, seed],
-        js=r"""(imgs, p, s, rs, gs, st) => {
+        js=r"""(imgs, p, s, rs, gs, st, ar, oc) => {
             const images = window.__uploadedImages || [];
             const b64Array = images.map(img => img.b64);
             const imgsJson = JSON.stringify(b64Array);