File size: 211 Bytes
c6f2c61
96387ab
 
2814d60
 
 
c6f2c61
2814d60
c6f2c61
2814d60
1
2
3
4
5
6
7
8
9
10
import os
import torch

import gradio as gr

def greet(name):
    return torch.cuda.device_count(), os.system("nvidia-smi")

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