import torch
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns

# Path to model
MODEL_PATH = "Model/EvoSEGA/evoSEGA.pt"

# Load YOLOv5 model
model = torch.hub.load("ultralytics/yolov5", "custom", path=MODEL_PATH, force_reload=True)

# Run inference
results = model("test.jpg")

# Print results
print(results.pandas().xyxy[0])  # Pandas dataframe
results.show()  # Show image with detections