Working but added inference times

app.py

@@ -80,7 +80,7 @@ with gr.Blocks() as iface:
     )
     gr.HTML()
     MainOutput = gr.TextArea()
-    CustomButtonInput = gr.Input()
+    CustomButtonInput = gr.TextArea()
     CustomButtonClassDiagram = gr.Button(CustomPrompts[0])
     CustomButtonPydotcode = gr.Button(CustomPrompts[1])
     CustomButtonClassDiagram .click(custom_generate_response, inputs=[CustomButtonInput, CustomPrompts[0]], outputs=MainOutput)