Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 from ultralytics import YOLO
 
 # โหลดโมเดล YOLOv8 ที่ฝึกมาเอง
-model = YOLO('best_V5.pt')  # เปลี่ยน 'your_model.pt' เป็นโมเดลของคุณ
+model = YOLO('best_V5.pt')  # เปลี่ยน 'best_V5.pt' เป็นโมเดลของคุณ
 
 def predict(image):
     # ทำการทำนาย
@@ -15,12 +15,12 @@ def predict(image):
     # วาด bounding boxes และ labels บนภาพ
     for result in results:
         boxes = result.boxes.xyxy.cpu().numpy()
-        labels = result.names
         confidences = result.boxes.conf.cpu().numpy()
+        class_ids = result.boxes.cls.cpu().numpy()
         
-        for box, confidence in zip(boxes, confidences):
-            x1, y1, x2, y2 = map(int, box)
-            label = labels[box[5]]  # Assuming the label is stored in the last column (index 5)
+        for box, confidence, class_id in zip(boxes, confidences, class_ids):
+            x1, y1, x2, y2 = map(int, box[:4])
+            label = result.names[int(class_id)]  # ดึง label จาก class_id
 
             # วาด bounding box
             cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
@@ -34,4 +34,4 @@ def predict(image):
     return pil_image
 
 demo = gr.Interface(fn=predict, inputs=gr.Image(type="numpy"), outputs="image")
-demo.launch()
+demo.launch()