demo / app.py
Nowah's picture
Create app.py
b3b56c2
raw
history blame contribute delete
348 Bytes
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()