1225

app.py

@@ -1,12 +1,23 @@
-import torch
-from ultralytics import YOLO
 import gradio as gr
+from ultralytics import YOLO
+import torch
+
+model = YOLO("best0715.pt") 
 
-model = YOLO("best0709.pt")  # 或者是從 hf_hub_download 取得
+def detect(image, conf_threshold=0.3):
+    results = model(image, conf=conf_threshold)
+    if len(results) == 0 or results[0].boxes is None or len(results[0].boxes) == 0:
+        return image, "No bear detected."
+    plotted = results[0].plot()
+    conf = results[0].boxes.conf[0].item()
+    return plotted, f"Detected bear with confidence {conf:.2f}"
 
-def detect(image):
-    results = model(image)
-    return results[0].plot()  # 回傳帶框圖像
+iface = gr.Interface(
+    fn=detect,
+    inputs=[gr.Image(type="pil"), gr.Slider(0.1, 0.9, value=0.3, label="Confidence Threshold")],
+    outputs=[gr.Image(), gr.Textbox()],
+    title="Taiwanese Black Bear Detector",
+    description="Upload an image for detection."
+)
 
-iface = gr.Interface(fn=detect, inputs="image", outputs="image")
-iface.launch(show_api=True)
+iface.launch(share=True)

best0709.pt → best0715.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1fb10d8bcd3c4556db7ccd050d39fceaea9fd8f56ec988ca2471bd27354633aa
-size 22525539
+oid sha256:06e2747470091835b8bfd96a083be14c5f6d9edea180fddb68a65563f39d50ef
+size 22513955