File size: 166 Bytes
96387ab
 
2814d60
 
 
fdfae9f
2814d60
 
 
1
2
3
4
5
6
7
8
9
import torch

import gradio as gr

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

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