import torch
import os

os.environ["CUDA_VISIBLE_DEVICES"]=""

model = torch.hub.load('ultralytics/yolov5', 'custom', path_or_model='model/yolov5l.pt')
model.eval()

def get_prediction_l(image):

    # Inference
    with torch.no_grad():
        results = model(image)

    # Results
    results.print()

    # Data
    pred_res = results.xyxy[0].tolist()

    return pred_res