Update app.py

app.py

@@ -91,7 +91,7 @@ def compress_image(image, quality=50):
     compressed_image = Image.open(buffer)
     return compressed_image
 
-def query(prompt, is_realistic, num_inference_steps, negative):
+def query(prompt, is_realistic, num_inference_steps, width, height):
     if prompt == "" or prompt == None:
         return None
 
@@ -103,7 +103,7 @@ def query(prompt, is_realistic, num_inference_steps, negative):
     payload = {
         "inputs": prompt,
         "seed": random.randint(1, 1000000000),
-        "parameters": {"num_inference_steps": num_inference_steps, "width": 480, "height": 640, "negative_prompt": negative}
+        "parameters": {"num_inference_steps": num_inference_steps, "width": width, "height": height}
     }
 
     response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
@@ -169,12 +169,15 @@ with gr.Blocks(theme='gstaff/xkcd', css=css) as app:
                 negative_prompt = gr.Textbox(label="Negative prompt", placeholder="Enter a negative prompt here", lines=2, elem_id="prompt-negative-input", value="")
                 is_realistic = gr.Checkbox(label="Realistic filter", value=True)
                 num_inference_steps = gr.Slider(label="Number of inference steps", minimum=1, maximum=50, step=1, value=4)
+                with gr.Row():
+                    width = gr.Slider(label="Width", minimum = 128, maximum = 1024, step = 32, value = 480)
+                    height = gr.Slider(label="Height", minimum = 128, maximum = 1024, step = 32, value = 640)
                     
         with gr.Row():
             text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
         with gr.Row():
             image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
         
-        text_button.click(query, inputs=[text_prompt, is_realistic, num_inference_steps, negative_prompt], outputs=image_output)
+        text_button.click(query, inputs=[text_prompt, is_realistic, num_inference_steps, width, height], outputs=image_output)
 
 app.launch(show_api=True, share=True)