import gradio as gr

def fake_predict(img):
    return {
        "女性醫護": 0.25,
        "男性醫護": 0.25,
        "女性工程師": 0.25,
        "男性工程師": 0.25
    }

# Gradio UI
demo = gr.Interface(
    fn=fake_predict,
    inputs=gr.Image(type="pil"),
    outputs=gr.Label(num_top_classes=4),
    title="性別平等職業分類器",
)

demo.launch()