Update app.py

app.py

@@ -55,7 +55,13 @@ def process_video(video):
             for box in boxes:
                 x1, y1, x2, y2 = map(int, box.xyxy[0].tolist())
                 class_id = int(box.cls[0])  # Class index for best.pt
-                label = f"Best: {best_classes[class_id]} - {box.conf[0]:.2f}"  # Map class_id to specific labels
+                
+                # Check if the class_id is within the range of best_classes
+                if class_id < len(best_classes):
+                    label = f"Best: {best_classes[class_id]} - {box.conf[0]:.2f}"  # Map class_id to specific labels
+                else:
+                    label = f"Best: Unknown Class - {box.conf[0]:.2f}"  # Handle out of range class_id
+
                 cv2.rectangle(frame, (x1, y1), (x2, y2), (255, 0, 0), 2)
                 cv2.putText(frame, label, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (255, 0, 0), 2)