File size: 218 Bytes
43b8d8c
f948f08
 
43b8d8c
b0f6b93
43b8d8c
f948f08
43b8d8c
 
 
1
2
3
4
5
6
7
8
9
10
11
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()