import gradio as gr import spaces import torch @spaces.GPU def greet(name): return f"Tensor: {torch.Tensor([0]).cuda()}, Name: {name}" iface = gr.Interface(fn=greet, inputs="text", outputs="text") iface.launch()