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()