File size: 348 Bytes
b3b56c2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | 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()
|