c6f2c61
96387ab
2814d60
c6f2c61
2814d60
c6f2c61
2814d60 |
|
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() |