File size: 220 Bytes
43b8d8c
f948f08
 
43b8d8c
205ed8a
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()