import spaces import gradio as gr @spaces.GPU() def test_fn(): return "Hello" with gr.Blocks() as demo: btn = gr.Button("Test") out = gr.Textbox() btn.click(test_fn, outputs=out)