unable to use best.pt to recognize real time images from screen

[ThiagoBauken]

i cant use my trained best.py to run with this locally but it runs on the pretreined version
import torch
import time
import mss
import numpy as np
import cv2

class MssYoloDetector:
def init(self):
# Carrega o modelo YOLOv5 pré-treinado
self.model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
# Ajuste para a resolução desejada: 1920x1080
self.monitor = {'top': 0, 'left': 0, 'width': 1920, 'height': 1080}

def run(self):
    with mss.mss() as sct:
        while True:
            screenshot = sct.grab(self.monitor)
            img = np.array(screenshot)
            img = cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)

            # Detecção com YOLO
            results = self.model(img)
            results.render()

            # Mostra o resultado
            cv2.imshow("Detecção YOLO", img)

            # Pressione 'q' para sair
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break

            time.sleep(0.01)

    cv2.destroyAllWindows()

if name == "main":
detector = MssYoloDetector()
detector.run()