Update main.py

main.py

@@ -176,6 +176,8 @@ def generate(
         payload = {
             "inputs": prompt ,
             "is_negative": negative_prompt,
+            "width": width,
+            "height": height,
             "steps": num_inference_steps,
             "cfg_scale": guidance_scale,
             "seed": seed if seed is not None else random.randint(-1, 2147483647)