Firearm_Detection_Yolov8n

pip install ultralytics huggingface_hub

from ultralytics import YOLO
from huggingface_hub import hf_hub_download

# Download model from Hugging Face Hub
model_path = hf_hub_download(
    repo_id="Subh775/Firearm_Detection_Yolov8n",
    filename="weights/best.pt"
)

# Load model
model = YOLO(model_path)

# Run inference
results = model("path/to/your/image.jpg")

# Display results
for box in results[0].boxes:
    class_name = model.names[int(box.cls[0])]
    confidence = box.conf[0]
    print(f"Detected: {class_name} (Confidence: {confidence:.3f})")

# Show annotated image
results[0].show()