Create app.py

app.py

@@ -0,0 +1,45 @@
+import gradio as gr
+from ultralytics import YOLO
+
+model = YOLO("best.pt")
+
+class_zh = {
+    "cardboard": "紙板 / 紙箱類",
+    "glass": "玻璃類",
+    "metal": "金屬類",
+    "paper": "紙類",
+    "plastic": "塑膠類",
+    "trash": "一般垃圾"
+}
+
+def predict_garbage(image):
+    results = model.predict(image, verbose=False)
+    result = results[0]
+
+    class_id = result.probs.top1
+    confidence = result.probs.top1conf.item()
+    class_name = result.names[class_id]
+
+    chinese_name = class_zh.get(class_name, class_name)
+
+    output_text = f"""
+預測類別：{class_name}
+
+中文說明：{chinese_name}
+
+信心分數：{confidence:.4f}
+
+信心百分比：{confidence * 100:.2f}%
+"""
+
+    return output_text
+
+demo = gr.Interface(
+    fn=predict_garbage,
+    inputs=gr.Image(type="pil", label="上傳垃圾圖片"),
+    outputs=gr.Textbox(label="模型預測結果"),
+    title="AI 垃圾分類影像辨識系統",
+    description="使用 YOLOv8 模型辨識垃圾圖片類別，包含 cardboard、glass、metal、paper、plastic、trash 六種類別。"
+)
+
+demo.launch()