import gradio as gr

def square(x):
    return x**2

demo = gr.Interface(
    fn=square,
    inputs=gr.Number(),
    outputs=gr.Number(),
    title="My First Demo"
)

demo.launch()