Update app.py

app.py

@@ -15,11 +15,11 @@ with gr.Blocks(theme='gradio/soft') as demo:
     with gr.Tab("Stable Diffusion"):
         prompt_input_txt = gr.Textbox(label="Prompt")
         negative_prompt_txt = gr.Textbox(label="Negative Prompt")
-        sd_input = gr.Image(label="Input Image")
-        sd_output = gr.Image(label="Output Image")
+        input_image = gr.Image(label="Input Image")
+        output_image = gr.Image(label="Output Image")
         submit_btn = gr.Button(value = "Submit")
-        sd_inputs = [prompt_input_seg, negative_prompt_seg, sd_input]
-        submit_btn.click(fn=infer_sd, inputs=sd_inputs, outputs=[sd_output])
+        sd_inputs = [prompt_input_txt, negative_prompt_txt, input_image]
+        submit_btn.click(fn=infer_sd, inputs=sd_inputs, outputs=[output_image])
     #examples = [["a woman is making an ok sign in front of a painting", "low quality", "val_1.png", "val_2.png" ]]
     #gr.Examples(fn = infer_sd, inputs = ["text", "text", "image", "image"], examples=examples, cache_examples=True)