Update app.py

app.py

@@ -23,13 +23,13 @@ with gr.Blocks() as demo:
         canny_input = gr.Image(label="Image")
         submit_btn = gr.Button("Submit")
         canny_inputs = [prompt_input_canny, negative_prompt_canny, canny_input]
-        submit_btn.click(fn=write_to_dataset, inputs=canny_inputs, outputs="image")
+        submit_btn.click(fn=infer_canny, inputs=canny_inputs, outputs="image")
     with gr.Tab("ControlNet with Semantic Segmentation"):
         prompt_input_seg = gr.Textbox(label="Prompt")
         negative_prompt_seg = gr.Textbox(label="Negative Prompt")
         seg_input = gr.Image(label="Image")
         submit_btn = gr.Button("Submit")
         seg_inputs = [prompt_input_seg, negative_prompt_seg, seg_input]
-        submit_btn.click(fn=write_to_dataset, inputs=seg_inputs, outputs="image")
+        submit_btn.click(fn=infer_segmentation, inputs=seg_inputs, outputs="image")
 
 demo.launch()