Update handler.py

handler.py

@@ -8,8 +8,8 @@ from io import BytesIO
 
 class EndpointHandler():
     def __init__(self, path=""):
-        hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/best.pt")
-        self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/best.pt", local_files_only=True))
+        hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/last.pt")
+        self.model = YOLO(hf_hub_download(repo_id="Drazcat-AI/flejes", filename="yolov8_flejes/runs/detect/train/weights/last.pt", local_files_only=True))
     
     def predict_objects(self, image_path):
         #results = self.model(image_path, imgsz=1280)