File size: 349 Bytes
d581231
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import os

import gradio as gr


def greet(name):
    if name == "cpu":
        cpu = os.environ.get("CPU_CORES")
        memory = os.environ.get("MEMORY")
        return f"CPU: {cpu} Memory: {memory}"
    if name == "gpu":
        return os.popen('nvidia-smi').read()


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