import gradio as gr
import torch

def greet(name):
    return torch.cuda.device_count()

demo = gr.Interface(fn=greet, inputs="text", outputs="text")

demo.launch()