主要

app.py

@@ -4,19 +4,22 @@ import gradio as gr
 # 載入模型
 learn = load_learner('gender_model.pkl')
 
+# 分類標籤
 labels = learn.dls.vocab
 
+# 預測函式
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
-# 建立介面
+# 建立 Gradio 介面
 demo = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),
     outputs=gr.Label(num_top_classes=4),
-    title="SDG5 性別分類模型",
-    description="上傳一張職業圖片，模型會預測屬於哪一類別（例如：女性醫護、男性醫護等）"
+    title="SDG5 性別平等影像分類模型",
+    description="上傳一張職業相關圖片，模型將預測其類別：女性醫護、男性醫護、女性工程師、男性工程師。"
 )
 
+# 執行應用
 demo.launch()