from ultralytics import YOLO

print("START")

model = YOLO("yolov8n.pt")

model.info()
print(model.info())

result = model("image_fire_2.jpg")

for r in result:
    for box in r.boxes:
        class_id = int(box.cls[0])
        confidence = float(box.conf[0])
        class_name = model.names[class_id]

        print(f"{class_name} - {confidence:.2%}")

result[0].save("output_4.jpg")