import gradio as gr
import spaces
import torch

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


@spaces.GPU
def slow_echo(message, history):
    return message


demo = gr.ChatInterface(slow_echo).queue().launch()