inference.py

from ultralytics import YOLO
import torch

def run_inference(image_path):
    model = YOLO('model.pt')
    results = model(image_path)
    return results