Update app.py

app.py

@@ -12,6 +12,8 @@ def predict(image):
     # ทำการทำนาย
     results = model(image)
 
+    detected = False  # ตัวแปรเพื่อตรวจสอบว่ามีการตรวจพบหรือไม่
+
     # วาด bounding boxes และ labels บนภาพ
     for result in results:
         boxes = result.boxes.xyxy.cpu().numpy()
@@ -29,6 +31,11 @@ def predict(image):
             label_text = f"{label} {confidence:.2f}"
             cv2.putText(image, label_text, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
 
+            detected = True  # ตั้งค่าตัวแปรเป็น True ถ้ามีการตรวจพบ
+
+    if not detected:  # ถ้าไม่มีการตรวจพบ
+        cv2.putText(image, "No detected", (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
+
     # แปลงภาพกลับเป็นรูปแบบที่ Gradio สามารถแสดงได้
     pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
     return pil_image