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):
+def query(prompt, is_realistic, num_inference_steps):
     if prompt == "" or prompt == None:
         return None
 
@@ -102,7 +102,8 @@ def query(prompt, is_realistic):
     
     payload = {
         "inputs": prompt,
-        "seed": random.randint(1, 1000000000)
+        "seed": random.randint(1, 1000000000),
+        "parameters": {"num_inference_steps": num_inference_steps}
     }
 
     response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
@@ -166,12 +167,13 @@ with gr.Blocks(theme='gstaff/xkcd', css=css) as app:
             with gr.Column(elem_id="prompt-container"):
                 text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=2, elem_id="prompt-text-input", value=test_prompt)
                 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():
             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], outputs=image_output)
+        text_button.click(query, inputs=[text_prompt, is_realistic, num_inference_steps], outputs=image_output)
 
 app.launch(show_api=True, share=True)