import gradio as gr
import spaces
import torch

zero = torch.Tensor([0]).cuda()
print(zero.device) # <-- 'cpu' 🤔

@spaces.GPU(duration=1)
def greet(n, name):
    print(name)
    n += 1
    return n, n

def greet_update(n, name):
    print(name)
    n += 1
    return gr.update(value=n), n

counter = gr.State(value=0)

gr.Interface(
    fn=greet,
    inputs=[counter, gr.Text()],
    outputs=[counter, gr.Number()],
).launch()